根据一个csv文件中的列信息从另一个csv文件中提取行

import csv

def extract_rows(column_info_file, data_file, output_file):
    # 读取列信息的CSV文件
    with open(column_info_file, 'r') as column_file:
        column_reader = csv.reader(column_file)
        column_info = next(column_reader)  # 假设列信息在第一行

    # 确定需要提取的列的索引
    extract_columns = [column_info.index(column) for column in column_info]

    # 读取待提取行的CSV文件
    with open(data_file, 'r') as data_file:
        data_reader = csv.reader(data_file)
        extracted_rows = []

        # 提取行数据
        for row in data_reader:
            extracted_row = [row[column] for column in extract_columns]
            extracted_rows.append(extracted_row)

    # 写入提取结果到新的CSV文件
    with open(output_file, 'w', newline='') as output:
        writer = csv.writer(output)
        writer.writerows(extracted_rows)

# 示例用法
column_info_file = 'column_info.csv'
data_file = 'data.csv'
output_file = 'extracted_rows.csv'

extract_rows(column_info_file, data_file, output_file)