熊猫数据框子图按列分组

Question

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

我想按组[0，1]和[2]列进行子图绘制。有办法吗？

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')