python csv文件取特定列

要从CSV文件中提取特定列，可以使用Python的CSV模块来实现。下面是一个示例代码，演示如何提取特定列：

import csv

def extract_column(csv_file, column_name):
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        if column_name not in reader.fieldnames:
            print(f"列名 '{column_name}' 在CSV文件中不存在。")
            return

        extracted_data = []
        for row in reader:
            extracted_data.append(row[column_name])

    return extracted_data

# 使用示例
csv_file = 'data.csv'
column_name = 'Name'
extracted_column = extract_column(csv_file, column_name)
print(extracted_column)

在上述代码中，我们定义了一个extract_column函数，它接受CSV文件路径和列名作为参数。函数首先打开CSV文件，并使用csv.DictReader创建一个读取器对象。然后，它检查指定的列名是否存在于CSV文件的标题行中，如果不存在，就会打印一条错误消息并返回。接下来，它遍历CSV文件的每一行，提取指定列的值，并将其存储在extracted_data列表中。最后，函数返回提取的列数据。

你只需将csv_file变量设置为你要读取的CSV文件的路径，将column_name变量设置为你要提取的列名，运行代码即可。提取的列数据将存储在extracted_column变量中，并打印出来。

请确保将csv_file的值替换为你实际的CSV文件路径，并提供你要提取的列的正确名称。