如何有效绘制matplotlib的代码

Posted 2021-05-04

我是python的新手，并进行了股票的时间序列分析。我根据收盘价的百分比变化创建了5只股票的滚动平均值的数据框。因此，该df有5列，而我有另一个df指数滚动平均值我想用索引df绘制df的单个股票列。我写了这段代码

fig.add_subplot(5,1,1)
plt.plot(pctchange_RA['HUL'])
plt.plot(N50_RA)    

fig.add_subplot(5,1,2)
plt.plot(pctchange_RA['IRCON'])
plt.plot(N50_RA)    

fig.add_subplot(5,1,3)
plt.plot(pctchange_RA['JUBLFOOD'])
plt.plot(N50_RA)    

fig.add_subplot(5,1,4)
plt.plot(pctchange_RA['PVR'])
plt.plot(N50_RA)    

fig.add_subplot(5,1,5)
plt.plot(pctchange_RA['VOLTAS'])
plt.plot(N50_RA)   


NOTE:pctchange_RA is a pandas df of 5 stocks and N50_RA is a index df of one column

答案

您可以将列名放在列表中，然后在其上循环并动态创建子图。伪代码如下所示

cols = ['HUL', 'IRCON', 'JUBLFOOD', 'PVR', 'VOLTAS'] for i, col in enumerate(cols): ax = fig.add_subplot(5, 1, i+1) ax.plot(pctchange_RA[col]) ax.plot(N50_RA)