用于循环csv头的Python字典用于绘图

时间:2017-04-10 10:56:00

标签: python dictionary key

Python新手在这里,我的目标是将csv加载到数据框中,这样我就可以遍历每个标题来绘制单个发行版。

我想使用字典将标题作为键和数据存储为值列表,如下所示:

header1: [1,2,3,4,5]
header2: [6,7,8,9,10]

然后我想遍历我的header_list中的所有标题(不确定这是否是最好的方法)来绘制。

PATH="/Users/Myname/Desktop/PFA/"
df=pd.read_csv(PATH + 'my_input.csv', sep=',',header=None)

#2nd row in csv is headers
header=df.iloc[[1]]

fom = dict()

for i in header_list:
     for key, value in my_dict.iteritems():
         plt.plot(fom[key])
         plt.show()

感谢任何帮助。你可以说我不知道​​我在做什么。

1 个答案:

答案 0 :(得分:0)

你可以更好地遍历df.columns

df = pd.read_csv(PATH + 'my_input.csv', sep=',', header=True)

# df.columns has the header
for column in df.columns:
    plt.plot(df[column], label=column)
plt.show()