Pandas read_csv() 函数设置列名指南

Pandas read_csv() 函数设置列名指南

pandas.read_csv() 函数的 header 参数可以用来指定数据文件的列名。

1. 手动指定列名

如果数据文件中没有列名，可以设置 header=None，然后手动指定列名：

import pandas as pd

df = pd.read_csv('data.csv', header=None, names=['id', 'name', 'age', 'gender'])

2. 使用第一行作为列名

如果数据文件中已经有列名，可以设置 header=0，表示使用第一行作为列名：

import pandas as pd

df = pd.read_csv('data.csv', header=0)

3. 指定其他行作为列名

如果数据文件中的列名不是第一行，可以设置 header=n，其中 n 为列名所在的行数（从 0 开始计数）：

import pandas as pd

df = pd.read_csv('data.csv', header=2)