Python批量提取多个csv文件内指定行数据并汇总在新csv文件内
以下是一个Python脚本示例,用于从多个CSV文件中提取指定行的数据,并将结果保存到一个新的CSV文件中。
import os
import csv
from glob import glob
# 设置源CSV文件路径、目标CSV文件路径和需要提取的行索引
source_csv_folder = 'path/to/csv/files' # CSV文件所在文件夹路径
target_csv_file = 'path/to/output.csv' # 目标CSV文件路径
rows_to_extract = [0, 2] # 需要提取的行索引列表,例如提取第0行和第2行
# 从所有CSV文件中提取指定行,并保存到目标CSV文件
with open(target_csv_file, 'w', newline='') as out_csv:
csv_writer = csv.writer(out_csv)
for file_path in glob(os.path.join(source_csv_folder, '*.csv')):
with open(file_path, 'r') as in_csv:
csv_reader = csv.reader(in_csv)
for row_index, row in enumerate(csv_reader):
if row_index in rows_to_extract:
csv_writer.writerow(row)
确保将source_csv_folder
变量设置为包含CSV文件的文件夹路径,target_csv_file
设置为要创建的新CSV文件的路径,rows_to_extract
设置为你想要提取的行索引列表。
此脚本会遍历指定文件夹内所有CSV文件,提取指定的行,并将结果保存到target_csv_file
指定的新CSV文件中。
评论已关闭