Question

我最近开始学习Pandas，在使用 groupby 和 agg 之后，我在如何绘制结果方面遇到了一些麻烦。使用Pandas，我创建了一个数据框，并根据两列＆＃39; ID＆＃39;和＆＃39; x＆＃39;。然后我从组中选择了一个特定的列（＆＃39;结果＆＃39;）来计算sem和mean。

具体来说，代码：

df = pd.read_csv('pandas_2015-11-7.csv')              
df_group = df.groupby(['x','ID'])['results']          
df_group_results = df_group.agg([stats.sem, np.mean])

结果如下所示：

            sem      mean
x    ID                    
2.5  0     0.010606  0.226674
     1     0.000369  0.490820
     2     0.000508  0.494094
5.0  0     0.001672  0.005059
     1     0.012252  0.190962
     2     0.003696  0.170342
7.5  0     0.001630  0.004506
     1     0.002567  0.016109
     2     0.002081  0.047301
10.0 0     0.000000  0.000000
     1     0.000000  0.000000
     2     0.000000  0.000000
12.5 0     0.000000  0.000000
     1     0.000000  0.000000
     2     0.000000  0.000000

我的问题是如何基于这些结果制作带误差线的线图？ x轴应该基于＆＃39; x＆＃39;价值和＆＃39; ID＆＃39;确定线条（在本例中为3行，图例为0,1和2）。我希望实现的理想情节如下plot http://matplotlib.org/1.2.1/mpl_examples/pylab_examples/errorbar_demo_00.png。

Answer 1

groupby（）方法返回层次索引（多索引）：

http://pandas.pydata.org/pandas-docs/stable/advanced.html

如果我创建一个具有类似层次索引的df：

import pandas as pd
df = pd.DataFrame({'mean':[0.5,0.25,0.7,0.8],'sem':[0.1,0.1,0.1,0.2]})
df.index = pd.MultiIndex(levels=[[2.5,5.0],[0,1]],labels=[[0,0,1,1],[0,1,0,1]],names=['x','ID'])

然后我有以下df：

        mean  sem
x   ID           
2.5 0   0.50  0.1
    1   0.25  0.1
5.0 0   0.70  0.1
    1   0.80  0.2

我可以从多索引中获取相关信息，并使用它来按顺序选择和绘制正确的行：

x_values = df.index.levels[0]
ID_values = df.index.levels[1]

for ID in ID_values:
    mean_data = df.loc[[(x,ID) for x in x_values],'mean'].tolist()
    error_data = df.loc[[(x,ID) for x in x_values],'sem'].tolist()
    matplotlib.pyplot.errorbar(x_values,mean_data,yerr=error_data)

legend(ID_values)

如何在Pandas的groupby之后绘制聚合结果？

1 个答案: