如何提取带有特定头名的CSV列并输出到新文件？

import csv
import pandas as pd

def extract_csv_column(csv_file, output_file, header_name):
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        headers = next(reader)  # 获取头名列表

        if header_name in headers:
            column_index = headers.index(header_name)  # 获取特定头名的索引
            data = pd.read_csv(csv_file)  # 使用pandas读取CSV文件
            column_data = data.iloc[:, column_index]  # 提取特定列数据

            with open(output_file, 'w', newline='') as output:
                writer = csv.writer(output)
                writer.writerow([header_name])  # 写入特定头名到新文件
                writer.writerows(zip(column_data))  # 写入提取的列数据到新文件

            print("提取成功！请查看新文件：", output_file)
        else:
            print("特定头名不存在于CSV文件中！")

# 示例用法
csv_file = "input.csv"  # 原始CSV文件路径
output_file = "output.csv"  # 新文件路径
header_name = "特定头名"  # 要提取的特定头名

extract_csv_column(csv_file, output_file, header_name)