如何在 Python 中获取一个 CSV 列

在 Python 中操作 CSV 文件非常常见，本文将介绍如何在 Python 中获取一个 CSV 文件的列数据。

读取 CSV 文件

首先，我们需要读取 CSV 文件。可以使用 Python 内置模块 csv 来读取 CSV 文件。下面是一个简单的例子：

import csv

# 打开 CSV 文件
with open('file.csv', 'r', encoding='utf-8') as csvfile:
    # 使用 csv.reader 读取 CSV 文件
    reader = csv.reader(csvfile)

    # 遍历 CSV 文件中的每一行数据
    for row in reader:
        print(row)

上面的代码会输出 CSV 文件的每一行数据。

获取 CSV 列

获取 CSV 列，可以使用下标或者字典的方式。下面是两个例子：

通过下标获取

import csv

# 打开 CSV 文件
with open('file.csv', 'r', encoding='utf-8') as csvfile:
    # 使用 csv.reader 读取 CSV 文件
    reader = csv.reader(csvfile)

    # 获取第二列数据
    col2 = [row[1] for row in reader]

    print(col2)

上面的代码会输出 CSV 文件的第二列数据。

通过字典获取

import csv

# 打开 CSV 文件
with open('file.csv', 'r', encoding='utf-8') as csvfile:
    # 使用 csv.DictReader 读取 CSV 文件
    reader = csv.DictReader(csvfile)

    # 获取 'age' 列数据
    age = [row['age'] for row in reader]

    print(age)

上面的代码会输出 CSV 文件的 'age' 列数据。使用字典获取列数据需要注意 CSV 文件的第一行必须是列名。

总结

本文介绍了如何在 Python 中获取一个 CSV 文件的列数据。使用 csv 模块可以方便的读取和操作 CSV 文件。获取列数据可以通过下标或者字典方式，使用时需要注意列名和下标的对应关系。