python按固定列拆分csv文件
时间: 2023-08-09 11:10:18 浏览: 82
如果你需要按照固定的列数进行拆分 CSV 文件,你可以使用 Python 自带的 csv 模块来读取和处理 CSV 文件,然后将每个拆分后的子文件中的行数写入指定的文件中。下面是一个示例代码:
```python
import csv
# 读取 CSV 文件
with open('input.csv', newline='') as csvfile:
reader = csv.reader(csvfile)
# 获取 CSV 表头
headers = next(reader)
# 设置拆分的列数
split_columns = 3
# 设置每个拆分后的子文件中的行数
split_size = 1000
# 创建拆分后的 CSV 文件
split_files = [open(f'output_{i}.csv', 'w', newline='') for i in range(split_columns)]
writers = [csv.writer(file) for file in split_files]
# 写入 CSV 表头
for writer in writers:
writer.writerow(headers)
# 拆分 CSV 文件
count = 0
for row in reader:
for i in range(split_columns):
writers[i].writerow([row[j] for j in range(i, len(row), split_columns)])
count += 1
# 每写入 split_size 行就将行数写入指定的文件中
if count % split_size == 0:
with open('rows_count.txt', 'a') as f:
f.write(f'{count}\n')
# 关闭拆分后的 CSV 文件
for file in split_files:
file.close()
```
上述代码将 `input.csv` 文件按照每 3 列进行拆分,并将拆分后的子文件命名为 `output_0.csv`、`output_1.csv` 和 `output_2.csv`。每个拆分后的子文件中最多包含 1000 行,当写入的行数达到 1000 行时,将行数写入 `rows_count.txt` 文件中。你可以根据自己的需求修改拆分的列数、拆分后的文件名和每个拆分后的子文件中的行数。
阅读全文