python将csv文件分割为两份
时间: 2024-06-16 19:07:18 浏览: 174
在Python中,可以使用csv模块来读取和处理CSV文件。要将CSV文件分割为两份,可以按照以下步骤进行操作:
1. 导入csv模块:
```python
import csv
```
2. 打开原始CSV文件和两个目标文件:
```python
with open('input.csv', 'r') as input_file, open('output1.csv', 'w', newline='') as output_file1, open('output2.csv', 'w', newline='') as output_file2:
reader = csv.reader(input_file)
writer1 = csv.writer(output_file1)
writer2 = csv.writer(output_file2)
```
3. 读取原始CSV文件的每一行数据,并根据需要将其写入到两个目标文件中:
```python
for row in reader:
# 根据某个条件判断将行写入到不同的文件中
if condition:
writer1.writerow(row)
else:
writer2.writerow(row)
```
在上述代码中,`condition`是一个用于判断行应该写入哪个文件的条件。
4. 关闭文件:
```python
input_file.close()
output_file1.close()
output_file2.close()
```
这样就可以将原始的CSV文件分割为两份,并保存到两个目标文件中。
阅读全文