使用熊猫读取csv的第0行并将其保存到列表中

Question

这是我的数据框：-

A            B            C           D
1            23           34          23
3            23           21           4
4            5            6            7

  

我想做的是读第0行并将其添加到像这样的列表[A,B,C,D]

     

（此列名称可以是任何东西）

我用过

data = pd.read_csv(file_path, nrows=0)

但是我无法将其添加/存储到列表中

该怎么做？

Answer 1

您可以轻松地做到这一点：

list(data)

但是更明确地写成这样：

list(data.columns.values)

编辑：正如RafAelC正确地说的那样，最好像这样使用tolist()：

data.columns.values.tolist()