Как эффективно построить этот код matplotlib - PullRequest
Новая
       2

Как эффективно построить этот код matplotlib

1 голос
/ 06 мая 2020

Я новичок в 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

1 Ответ

0 голосов
/ 06 мая 2020

Вы можете поместить имена столбцов в список, а затем просто добавить l oop поверх него и динамически создавать подзаголовки. Псевдокод будет выглядеть следующим образом: 

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