从csv文件中调出列的函数

import csv

def extract_column_from_csv(csv_file, column_name):
    # 打开csv文件
    with open(csv_file, 'r') as file:
        # 读取文件内容
        csv_data = csv.reader(file)
        
        # 解析csv数据
        headers = next(csv_data)  # 获取表头
        column_index = headers.index(column_name)  # 获取指定列的索引
        
        # 提取特定列的数据
        column_data = [row[column_index] for row in csv_data]
        
        # 返回提取的列数据
        return column_data

csv_file = 'data.csv'
column_name = '列名'

extracted_data = extract_column_from_csv(csv_file, column_name)
print(extracted_data)