使用 Pandas 获取数据列名列表的方法

使用 Pandas 获取数据列名列表的方法

在 Pandas 中，我们经常需要获取到数据集中的列名列表，以便进一步对数据进行处理。这里介绍两种获取列名列表的方式。

1. 使用.columns属性

Pandas.DataFrame 对象拥有一个名为 .columns 的属性，该属性返回一个包含所有列名的列表。

示例代码：

import pandas as pd

#读取csv文件
data = pd.read_csv('data.csv')

# 获取列名列表
cols = data.columns.tolist()

print(cols