使用for循环将来自不同CSV文件的列添加到新的CSV文件的步骤如下:
下面是一个示例代码,演示如何使用for循环将来自不同CSV文件的列添加到新的CSV文件:
import csv
import os
# 创建新的CSV文件
new_csv_file = open('new_file.csv', 'w', newline='')
new_csv_writer = csv.writer(new_csv_file)
# 获取包含要处理的CSV文件的目录中的所有文件名
csv_files = os.listdir('csv_files_directory')
for file_name in csv_files:
# 打开当前CSV文件
with open('csv_files_directory/' + file_name, 'r') as csv_file:
csv_reader = csv.reader(csv_file)
for i, row in enumerate(csv_reader):
if i == 0:
# 写入列标题
new_csv_writer.writerow(row)
else:
# 写入特定列数据
new_csv_writer.writerow([row[column_index] for column_index in [1, 3, 5]])
# 关闭当前CSV文件
csv_file.close()
# 关闭新的CSV文件
new_csv_file.close()
在上述示例代码中,我们假设要处理的CSV文件存储在名为"csv_files_directory"的目录中。代码将遍历该目录中的所有文件,并将每个文件的第2、第4和第6列数据写入新的CSV文件"new_file.csv"中。你可以根据实际需求修改代码中的文件路径和列索引。
领取专属 10元无门槛券
手把手带您无忧上云