使用熊貓讀取第 0 行 csv 並將其保存到列表中

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