python打开csv获取指定列

假设我们有一个名为"example.csv"的文件，它包含以下内容：

name,age,gender
John,25,Male
Lisa,30,Female
David,28,Male

我们可以使用Python的csv模块打开这个文件，并获取指定列的值。

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    # 获取第二列（索引为1）的所有值
    column_values = [row[1] for row in reader]
    print(column_values)

输出：

['age', '25', '30', '28']

我们发现第一行也被读取了，我们可以使用next()函数跳过第一行：

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    # 跳过第一行
    next(reader)
    # 获取第二列（索引为1）的所有值
    column_values = [row[1] for row in reader]
    print(column_values)

输出：

['25', '30', '28']

如果我们想要获取多列的值，可以简单地在列表推导式中添加索引：

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    # 跳过第一行
    next(reader)
    # 获取第一列和第三列的值
    column_values = [(row[0], row[2]) for row in reader]
    print(column_values)

输出：

[('John', 'Male'), ('Lisa', 'Female'), ('David', 'Male')]