Python 从 CSV 读取列

CSV（Comma-Separated Values）是一种常见的电子数据交换格式。在Python中，我们可以使用csv模块来操作CSV文件。

读取整个CSV文件

要读取整个CSV文件，我们可以使用csv模块的reader函数。这个函数返回一个由行列表示的迭代器，每一行都是一个由逗号分隔的值组成的列表。

import csv

with open('example.csv') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

上面的代码将打开名为example.csv的文件，并将它读入reader对象，然后使用for循环逐行遍历该文件并输出每一行。

读取指定列

如果我们只想读取CSV文件中的某些列，我们可以在上述代码的基础上进行修改。在下面的代码中，我们只会输出文件的第一和第三列。

import csv

with open('example.csv') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row[0], row[2])

pandas 模块的应用

另外，我们还可以使用pandas模块来操作CSV文件。pandas是Python中一个重要的数据分析库，其可以快速，方便地读取和操作CSV文件。

以下代码将演示如何用pandas读取CSV文件的某一列：

import pandas as pd

data = pd.read_csv('example.csv')
col1 = data['col1']
print(col1)

上述代码将打开名为example.csv的文件，并打印出名为col1的列。

总之，使用Python读取CSV文件中的列有多种方法，我们可以使用csv模块获取完整数据并提取指定列，也可以使用pandas模块进行更为高级的处理，以满足不同的需求。