Python 代码:提取多个 CSV 文件的特定列并创建新 CSV 文件
以下是将不同 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
原文地址: https://www.cveoy.top/t/topic/oEzI 著作权归作者所有。请勿转载和采集!