python - 将列名传递给 Pandas read_csv() 函数

Question

我有一个如下所示的 csv 文件:

,time,o,h,l,c
0,2021-03-17 09:30:00,101.25,103.98,90.17,101.78
1,2021-03-17 09:45:00,102.83,107.49,95.22,101.93
2,2021-03-17 10:00:00,100.92,102.90,97.19,102.80

我正在使用 read_csv 函数读取此数据，如下所示:

cols = ['indx', 'timestamp', 'open', 'high', 'low', 'close']
df = pd.read_csv('prices.csv', names=cols)

生成的数据框如下所示:

    Unnamed: 0  time                o       h       l       c
0   0           2021-03-17 09:30:00 101.25  103.98  90.17   101.78
1   1           2021-03-17 09:45:00 102.83  107.49  95.22   101.93
2   2           2021-03-17 10:00:00 100.92  102.90  97.19   102.80

传递 names 参数似乎对列名称​​没有影响。

编辑:

传递 header 参数(根据文档)导致相同的行为:

cols = ['indx', 'timestamp', 'open', 'high', 'low', 'close']
df = pd.read_csv('prices.csv', header=0, names=cols)

是否必须向 read_csv 函数传递另一个参数来自定义列名？

谢谢!

Answer 1

我的猜测是您需要将 header=0 传递给 read_csv:

df = pd.read_csv('prices.csv', names=cols, header=0)

输出:

>>> df
   indx            timestamp    open    high    low   close
0     0  2021-03-17 09:30:00  101.25  103.98  90.17  101.78
1     1  2021-03-17 09:45:00  102.83  107.49  95.22  101.93
2     2  2021-03-17 10:00:00  100.92  102.90  97.19  102.80