使用子图,是否有一种pythonic方法来绘制每个子图的多行?我有一个pandas数据框,有两个行索引,datestring和fruit,包含列的存储和值的数量。我想要5个子图,每个商店一个,日期字符串作为x轴,数量作为y轴,每个水果作为自己的彩色线。
df.plot(subplots=True)
我认为,几乎让我有适量的子图,除了它完全聚合数量而不是按果实绘图。
答案 0 :(得分:5)
<强> 设置 强>
始终提供可重现问题的样本数据
我在这里提供了一些
cols = pd.Index(['TJ', 'WH', 'SAFE', 'Walmart', 'Generic'], name='Store')
dates = ['2015-10-23', '2015-10-24']
fruit = ['carrots', 'pears', 'mangos', 'banannas',
'melons', 'strawberries', 'blueberries', 'blackberries']
rows = pd.MultiIndex.from_product([dates, fruit], names=['datestring', 'fruit'])
df = pd.DataFrame(np.random.randint(50, size=(16, 5)), rows, cols)
df
首先,您要将行索引的第一级转换为pd.to_datetime
df.index.set_levels(pd.to_datetime(df.index.levels[0]), 0, inplace=True)
现在我们可以看到我们可以直观地绘制
# fill_value is unnecessary with the sample data, but should be there
df.TJ.unstack(fill_value=0).plot()
我们可以用
绘制所有这些fig, axes = plt.subplots(5, 1, figsize=(12, 8))
for i, (j, col) in enumerate(df.iteritems()):
ax = axes[i]
col = col.rename_axis([None, None])
col.unstack(fill_value=0).plot(ax=ax, title=j, legend=False)
if i == 0:
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', ncol=1)
fig.tight_layout()