在pandas数据框中,如果我创建一个枢轴并基于该列进行绘图,则如何限制基于聚合函数绘制的列数。
示例
假设我有一个数据框枢纽
sum
name 'a' 'b' 'c' 'd'
key
1 1 2 3 4
2 1 2 3 4
3 1 2 3 nan
我绘制为
现在,我只想根据平均值绘制前(2)列,这样只有'c'和'd'会出现抑制'a'和'd'的情况。
如何使用熊猫来实现这一目标?
输入数据框和图
from io import StringIO
import pandas as pd
TESTDATA = StringIO("""name;key;value
'a';1;1
'a';2;1
'a';3;1
'b';1;2
'b';2;2
'b';3;2
'c';1;3
'c';2;3
'c';3;3
'd';1;4
'd';2;4
""")
df = pd.read_csv(TESTDATA, sep=";")
pivot = pd.pivot_table(df, index='key', columns='name', values='value',aggfunc=[np.sum])
pivot.plot()
答案 0 :(得分:3)