Pandas Dataframe groupby Display

Question

Lets say I have a DataFrame like following.

      A      B
0    text1   200
1    text2   200
2    text1   400
3    text2   500
4    text1   300
5    text1   600
6    text2   300

I want to print following output

     A             B
0    text1,text2   200
2    text1         400
3    text2         500
4    text1,text2   300
5    text1         600

There is no order,I just want to take text labels of column "A" for matching values in column "B". This has to be done with df.groupby as I know. Any way no success with my efforts yet. Hope you get my question.

Andy Hayden · Accepted Answer · 2013-08-05 11:12:48Z

9

You can nearly do this with a plain ol' sum:

In [11]: df.groupby('B').sum()
Out[11]:
              A
B
200  text1text2
300  text1text2
400       text1
500       text2
600       text1

You could use an aggregate with a join:

In [12]: df.groupby('B').agg(lambda x: ', '.join(x.values))
Out[12]:
                A
B
200  text1, text2
300  text1, text2
400         text1
500         text2
600         text1

answered Aug 5, 2013 at 11:12

Andy Hayden

378k110 gold badges640 silver badges546 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Nilani Algiriyage Over a year ago

As always your answers are short sweet and elegant..:) Thanks worked perfectly.

Andy Hayden Over a year ago

Also, nearly a transform: df.groupby('B').transform(', '.join) (which would have been more elegant)... Glad it helped!

Collectives™ on Stack Overflow

Pandas Dataframe groupby Display

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related