从pandas plot方法

Question

我正在绘制带有子图的熊猫数据框，因此我得到一个带有多个轴的np.array。

array([<matplotlib.axes._subplots.AxesSubplot object at blablabla>,
       <matplotlib.axes._subplots.AxesSubplot object at blablabla>,
       <matplotlib.axes._subplots.AxesSubplot object at blablabla>])

我想获取此输出以编辑标题，x标签并将其另存为pdf。如果它只是一个轴，我会首先获取变量.plot的输出，比如说ax，然后设置标题并用fig = ax.get_figure()得到数字以保存它的方式我想。我怎么能在这里做同样的事？

Answer 1

让我们使用df.plot前面的ax =获取轴列表。然后你可以使用列表切片来访问每个轴对象和set_title等，如下所示：

df = pd.DataFrame(np.random.randn(1000, 4), columns=list('ABCD'))
df = df.cumsum()

ax = df.plot(subplots=True)
ax[0].set_title('Series A')
ax[1].set_title('Series B')
ax[2].set_title('Series C')
ax[3].set_title('Series D')
fig = ax[0].get_figure()
fig.tight_layout()