groupby on a single column in Python Pandas returns a series. Want a single-column data frame?
Instead of
df.groupby('passenger_count')['total_amount'].mean()pass a one-element list of data columns:
df.groupby('passenger_count')[['total_amount']].mean()