将不同csv文件中的列提出并创建为一个新的csv文件的python代码及例子
以下是将不同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/hbDN 著作权归作者所有。请勿转载和采集!