Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Plotting a Pandas DataSeries.GroupBy

Tags:

I am new to python and pandas, and have the following DataFrame.

How can I plot the DataFrame where each ModelID is a separate plot, saledate is the x-axis and MeanToDate is the y-axis?

Attempt

data[40:76].groupby('ModelID').plot()

enter image description here

DataFrame

enter image description here

like image 829
Nyxynyx Avatar asked May 04 '13 15:05

Nyxynyx


People also ask

How do you get Groupby rows in pandas?

You can group DataFrame rows into a list by using pandas. DataFrame. groupby() function on the column of interest, select the column you want as a list from group and then use Series. apply(list) to get the list for every group.


2 Answers

You can make the plots by looping over the groups from groupby:

import matplotlib.pyplot as plt

for title, group in df.groupby('ModelID'):
    group.plot(x='saleDate', y='MeanToDate', title=title)

See for more information on plotting with pandas dataframes:
http://pandas.pydata.org/pandas-docs/stable/visualization.html
and for looping over a groupby-object:
http://pandas.pydata.org/pandas-docs/stable/groupby.html#iterating-through-groups

like image 51
joris Avatar answered Sep 20 '22 20:09

joris


Example with aggregation:

I wanted to do something like the following, if pandas had a colour aesthetic like ggplot:

aggregated = df.groupby(['model', 'training_examples']).aggregate(np.mean)
aggregated.plot(x='training_examples', y='accuracy', label='model')

(columns: model is a string, training_examples is an integer, accuracy is a decimal)

But that just produces a mess.

Thanks to joris's answer, I ended up with:

for index, group in df.groupby(['model']):
    group_agg = group.groupby(['training_examples']).aggregate(np.mean)
    group_agg.plot(y='accuracy', label=index)

I found that title= was just replacing the single title of the plot on each loop iteration, but label= does what you'd expect -- after running plt.legend(), of course.

like image 22
chbrown Avatar answered Sep 20 '22 20:09

chbrown