从熊猫数据透视表中选择一列

时间:2018-09-07 11:36:00

标签: python pandas pivot-table python-3.5

我有下面的数据透视表,它是使用以下代码从数据帧创建的:

table = pd.pivot_table(df, values='count', index=['days'],columns['movements'], aggfunc=np.sum)

movements 0    1   2   3   4   5   6   7
days
0         2777 51  2                    
1         6279 200 7   3                
2         5609 110 32  4                
3         4109 118 101 8   3            
4         3034 129 109 6   2   2        
5         2288 131 131 9   2   1        
6         1918 139 109 13  1   1        
7         1442 109 153 13  10  1        
8         1085 76  111 13  7           1
9         845  81  86  8   8            
10        646  70  83  1   2   1   1

从数据透视表中可以看到,它有0到7的8列,现在我想绘制一些特定的列而不是全部。我无法选择列。假设我要针对索引绘制第0列和第2列。 y应该使用什么来选择列0和列2?

plt.plot(x=table.index, y=??)

我尝试使用y = table.value['0', '2']y=table['0','2'],但没有任何效果。

1 个答案:

答案 0 :(得分:1)

如果您需要在单个绘图中使用这两个列值,则无法为y选择ndarray:

plt.plot(table['0'])
plt.plot(table['2'])

如果列名是整数,则:

plt.plot(table[0])
plt.plot(table[2])