python - pandas 数据框在 matplotlibm 中绘制为表格，但缺少第一列

Question

我想在 matplotlib 中将 pandas 数据框绘制为表格。

数据框(df):

  name   wait_value
  cscas   2
  cscas   6
  dgref   3
  dgref   7

我用过:

new_df = df.groupby(['name'])['wait_value'].mean().to_frame()
fig, axarr = plt.subplots(1, 2)
axarr[0].axis('off')
axarr[0].axis('tight')
axarr[0].table(cellText= new_df.values, colLabels=('ave_wait_value'), loc='center')
fig.tight_layout()

plt.show()

但是，该表只有一列“ave_wait_value”。我需要两列:

   ave_wait_value
     4
     5

如何将“名称”列添加到表中？

我也尝试过:

new_df.rename(columns = {0:'name', 1:'ave_wait_value'}, inplace =True)

但是，它不起作用。

另外，我需要计算每个“名称”的 std.dev 并将其作为同一表中的列，如何设计查询？ 谢谢!

Answer 1

名称信息就在那里；它只是存储在索引中。要将索引转换为列，请使用 reset_index:

In [45]: df.groupby(['name'])['wait_value'].mean().to_frame()
Out[45]:
       wait_value
name
cscas           4
dgref           5

In [46]: df.groupby('name')['wait_value'].mean().reset_index()
Out[46]:
    name  wait_value
0  cscas           4
1  dgref           5

要获取平均值和标准差，请使用agg:

In [49]: df.groupby('name')['wait_value'].agg(['mean', 'std']).reset_index()
Out[49]:
    name  mean       std
0  cscas     4  2.828427
1  dgref     5  2.828427