以下是将不同 CSV 文件中的指定列提取并创建为一个新的 CSV 文件的 Python 代码:

import csv

# 指定要提取的列
columns = ['name', 'age', 'gender']

# 创建一个新的 CSV 文件
with open('new_file.csv', 'w', newline='') as new_file:
    writer = csv.writer(new_file)
    writer.writerow(columns)

    # 遍历每个原始 CSV 文件
    for file_name in ['file1.csv', 'file2.csv', 'file3.csv']:
        with open(file_name, 'r') as file:
            reader = csv.DictReader(file)
            for row in reader:
                # 提取指定列的数据
                data = [row[column] for column in columns]
                writer.writerow(data)

假设原始 CSV 文件如下:

file1.csv:

name,age,gender,city
Alice,25,Female,New York
Bob,32,Male,San Francisco
Charlie,18,Male,Los Angeles

file2.csv:

name,age,gender,city
David,29,Male,Houston
Emily,22,Female,Boston
Frank,45,Male,Seattle

file3.csv:

name,age,gender,city
Grace,27,Female,Chicago
Henry,36,Male,Miami

那么运行上述 Python 代码之后,将会生成一个新的 CSV 文件 new_file.csv,内容如下:

name,age,gender
Alice,25,Female
Bob,32,Male
Charlie,18,Male
David,29,Male
Emily,22,Female
Frank,45,Male
Grace,27,Female
Henry,36,Male
Python 代码:提取多个 CSV 文件的特定列并创建新 CSV 文件

原文地址: https://www.cveoy.top/t/topic/oEzI 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录