使用 Python 从 CSV 获取列名

在处理 CSV 文件时，我们通常需要获取文件的列名信息。Python 提供了许多方法来实现这个目标。下面将介绍两种获取列名的方法。

方法一：使用 csv 模块

Python 的 csv 模块提供了一个 DictReader 类，它可以读取 CSV 文件并将每行转换成一个字典，键为列名，值为单元格内容。

import csv

with open('data.csv', 'r', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    headers = reader.fieldnames
    print(headers)

运行以上代码，将打开名为 data.csv 的文件，将其中的列名存储在 headers 变量中并打印出来。如果你的 CSV 文件第一行不是列名而是数据，则需要在打开文件时添加参数 skipinitialspace=True。

方法二：使用 pandas 库

Pandas 是一个非常强大的数据处理库，可以快速读取 CSV 文件并将其转换为数据框（DataFrame）。要读取文件的列名，我们只需要使用 read_csv 函数读取文件，然后使用数据框的 columns 属性获取列名。

import pandas as pd

df = pd.read_csv('data.csv')
headers = list(df.columns)
print(headers)

运行以上代码，将打开名为 data.csv 的文件，将其中的列名存储在 headers 变量中并打印出来。

以上两种方法都能够很好地获取 CSV 文件的列名信息。根据需要，选择合适的方法即可。