python读取csv文件按行分为两个文件
时间: 2023-11-18 14:51:27 浏览: 149
python读取csv文件指定行的2种方法详解
5星 · 资源好评率100%
可以使用Python的csv模块来读取csv文件,并按行将文件分割为两个文件。以下是一种实现方法:
```python
import csv
# 打开原始csv文件和两个新的csv文件
with open('A.csv', 'r') as csvfile, open('file1.csv', 'w', newline='') as file1, open('file2.csv', 'w', newline='') as file2:
reader = csv.reader(csvfile) # 创建一个reader对象
writer1 = csv.writer(file1) # 创建一个writer对象,用于写入file1.csv
writer2 = csv.writer(file2) # 创建一个writer对象,用于写入file2.csv
for i, row in enumerate(reader):
# 根据行数将记录写入不同的文件
if i < len(reader) // 2:
writer1.writerow(row)
else:
writer2.writerow(row)
```
这段代码将原始的csv文件按行分割为两个文件:file1.csv和file2.csv。前一半的记录将被写入file1.csv,后一半的记录将被写入file2.csv。
阅读全文