将变量名称读入列表

时间:2014-04-28 15:14:58

标签: python list pandas

鉴于已经读入数据帧的csv文件,有没有办法以编程方式将第0行(列/变量名称)读入列表中,以便以后使用?

我试图像这样迭代列:for i in cols: column = df[i]

1 个答案:

答案 0 :(得分:0)

只需在read_csv

中设置nrows=1即可
df = pd.read_csv(r'mycsv.csv', nrows=1)
cols = list(df.columns)