Using Pandas groupby to combine data [duplicate]

Question

I have a simple Pandas dataframe that I wish to groupby using a certain column. The df looks like the one below.

Color    Car
  R      Ford
  O      Kia
  Y      Mazda
  R      Chevrolet

I want to groupby 'Color', so the resulting df would be:

Color    Car
  R      Ford, Chevrolet 
  O      Kia
  Y      Mazda

This seems to be easy enough using pandas groupby. My code looks like the following:

df = df.groupby(['Color'])

But I get the following error:

Cannot access callable attribute 'iloc' of 'DataFrameGroupBy' objects, try using the 'apply' method

Why does groupby not work? It seems like the most basic operation for which groupby is the perfect thing to use?

Aside from your error, related: stackoverflow.com/questions/27298178/… — EdChum
– EdChum, Commented Apr 4, 2019 at 14:36

Chris Adams · Accepted Answer · 2019-04-04 14:38:59Z

2

Use:

df.groupby('Color')['Car'].apply(', '.join)

[out]

Color
O               Kia
R    Ford, Chevrolet
Y             Mazda

answered Apr 4, 2019 at 14:38

Chris Adams

18.7k4 gold badges26 silver badges44 bronze badges

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

1 Answer 1