pandas - 熊猫数据框子图按列分组

Question

df = pd.DataFrame([[0, 1, 2], [0, 1, 2]])
df.plot(subplots=True)

我想要按组 [0, 1] 和 [2] 列的子图。有办法吗？

标签： pandas

Answer 1

您可以DataFrameGroupBy.plot按Index.map字典使用 2 组：

mapping = {0:'a', 1:'a', 2:'b'}

df.groupby(df.columns.map(mapping.get), axis=1).plot()

详情：

print (df.columns.map(mapping.get))
Index(['a', 'a', 'b'], dtype='object')