[FIXED] Problem with a column in my groupby new object

Issue

so I have a dataframe and I made this operation:

df1 = df1.groupby(['trip_departure_date']).agg(occ = ('occ', 'mean'))

The problem is that when I try to plot, it gives me an error and it says that trip_departure_date doesn't exist!

I did this:

df1.plot(x = 'trip_departure_date', y = 'occ', figsize = (8,5), color = 'purple')

and I get this error:

KeyError: 'trip_departure_date'

Please help!

Solution

Your question is similar to this question: pandas groupby without turning grouped by column into index

When you group by a column, the column you group by ceases to be a column, and is instead the index of the resulting operation. The index is not a column, it is an index. If you set as_index=False, pandas keeps the column over which you are grouping as a column, instead of moving it to the index.

The second problem is the .agg() function is also aggregating occ over trip_departure_date, and moving trip_departure_date to an index. You don't need this second function to get the mean of occ grouped by trip_departure_date.

import pandas as pd

df1 = pd.read_csv("trip_departures.txt")

df1_agg = df1.groupby(['trip_departure_date'],as_index=False).mean()

Or if you only want to aggregate the occ column:

df1_agg = df1.groupby(['trip_departure_date'],as_index=False)['occ'].mean()

df1_agg.plot(x = 'trip_departure_date', y = 'occ', figsize = (8,5), color = 'purple')

Answered By - K. Thorspear