使用一长串数据帧头的Stackplot

时间:2018-07-12 16:44:47

标签: python pandas matplotlib

我正在使用以下代码创建堆积的折线图。现在,我正在从熊猫数据框中绘制数据。使用下面的代码,我为这个df中的变量提供了3个标头的列表,我对此感到满意,并且对此感到满意。但是,我想输入3个以上的变量(大约80个)。因此,我正在寻找一种可靠的循环方法,以避免在80行中的每行中键入内容,但是我还没有弄清楚。

有什么简单的想法可以实现这一目标吗?

谢谢!

import matplotlib.pyplot as plt
df['Date'] = pd.to_datetime(df['Date'])
df.index = df['Date'] 

y1 = df[headers_list[0]].resample('m').sum()
y2 = df[headers_list[1]].resample('m').sum()
y3 = df[headers_list[2]].resample('m').sum()
x = list(range(len(y1)))   
y = np.vstack([y1, y2, y3])
labels = [headers_list[0], headers_list[1], headers_list[2]]
fig, ax = plt.subplots(figsize=(20,6))
ax.stackplot(x, y1, y2, y3, labels=labels)
ax.legend(loc=2)
plt.show()

0 个答案:

没有答案