帮我写一个程序,遍历我输入的文件中中csv文件,将第一行列名删除,将前三列数据合并在第一列,用空格隔开,删除其他列数据,并以原文件名输出到我输出的文件夹中
时间: 2023-12-20 07:03:16 浏览: 61
你可以参考以下代码来实现:
```python
import os
import csv
# 输入文件夹路径
input_folder = "input_folder_path"
# 输出文件夹路径
output_folder = "output_folder_path"
# 遍历输入文件夹中的所有文件
for filename in os.listdir(input_folder):
if filename.endswith(".csv"):
# 构造输入文件路径和输出文件路径
input_file = os.path.join(input_folder, filename)
output_file = os.path.join(output_folder, filename)
# 打开输入文件和输出文件
with open(input_file, "r") as f_in, open(output_file, "w", newline="") as f_out:
# 读取输入文件中的CSV内容
reader = csv.reader(f_in)
# 写入输出文件中的CSV内容
writer = csv.writer(f_out)
# 跳过第一行(列名)
next(reader)
# 遍历每一行,进行处理
for row in reader:
# 合并前三列数据
merged_data = " ".join(row[:3])
# 删除其他列数据
del row[3:]
# 将合并后的数据插入到第一列
row.insert(0, merged_data)
# 写入到输出文件中
writer.writerow(row)
```
你需要将 `input_folder_path` 和 `output_folder_path` 替换成你自己的文件夹路径,并将代码放入一个 Python 文件中进行运行。
阅读全文