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

Question

this is my dataframe:-

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

what i want to do is read 0th row and add it into a list like this [A,B,C,D]

(here this column name can be anything)

I have used

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

But i am not able to add/store it into list

How to do that?

Answer 1

You can easily do this:

list(data)

but it is more explicit to write like t his:

list(data.columns.values)

Edit: As RafAelC said correctly it is better to use tolist() like t his:

data.columns.values.tolist()