python如何把一个文件夹中的所有csv文件的第一行输出到一个新的csv文件,第二行输出到第二个csv文件,以此类推
时间: 2024-02-20 14:00:01 浏览: 205
您可以使用Python中的csv模块和os模块来实现这个功能。下面是一份示例代码:
```python
import csv
import os
# 设置文件夹路径
folder_path = "your_folder_path"
# 获取所有csv文件的文件名
csv_files = [file for file in os.listdir(folder_path) if file.endswith('.csv')]
# 遍历所有csv文件
for i, file in enumerate(csv_files):
# 打开csv文件并读取第i行
with open(os.path.join(folder_path, file), 'r') as f:
reader = csv.reader(f)
rows = [row for j, row in enumerate(reader) if j == i]
# 将第i行写入新的csv文件
with open(f'output_{i+1}.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(rows)
```
这份代码将会读取指定文件夹路径下的所有csv文件,然后将它们的第1行依次写入`output_1.csv`,第2行依次写入`output_2.csv`,以此类推。注意,程序将会自动创建新的csv文件,所以您不需要手动创建这些文件。另外,如果您的csv文件中包含有表头,您可以将`rows = [row for j, row in enumerate(reader) if j == i]`这行代码修改为`rows = [row for j, row in enumerate(reader) if j == i+1]`,这样程序就会跳过表头并从第2行开始读取数据。
阅读全文