python - pandas read_csv 将 header 解析为字符串类型，但我想要整数

Question

例如，csv文件如下，(1,2,3)是标题!

1,2,3
0,0,0

我使用 pd.read_csv 读取 csv 文件并打印

import pandas as pd
df = pd.read_csv('./test.csv')
print(df[1])

发生错误关键错误:1

看起来read_csv将 header 解析为字符串..

有没有办法在数据帧列中使用整数类型？

Answer 1

我认为更一般的做法是将列名转换为整数 by astype :

df = pd.read_csv('./test.csv')
df.columns = df.columns.astype(int)

另一种方法是首先仅获取第一列并使用read_csv中的参数names :

import csv
with open("file.csv", "r") as f:
    reader = csv.reader(f)
    i = np.array(next(reader)).astype(int)

#another way
#i = pd.read_csv("file.csv", nrows=0).columns.astype(int)
print (i)
[1 2 3]

df = pd.read_csv("file.csv", names=i, skiprows=1)
print (df.columns)
Int64Index([1, 2, 3], dtype='int64')