python按照某一列拆分csv文件
时间: 2023-08-09 09:09:15 浏览: 275
自动拆分csv文件,并存储为utf-8格式的csv文件
你可以使用Python的csv模块来读取和处理csv文件,然后根据某一列的值来拆分文件。以下是一个示例代码,将根据第二列的值来拆分文件:
```python
import csv
# 读取csv文件
with open('input.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
# 读取表头
header = next(reader)
# 创建一个字典,用于存储拆分后的文件
split_files = {}
for row in reader:
# 获取第二列的值
col_value = row[1]
# 如果拆分文件不存在,则创建一个新的文件,并写入表头
if col_value not in split_files:
split_files[col_value] = open(col_value + '.csv', 'w', newline='')
writer = csv.writer(split_files[col_value])
writer.writerow(header)
# 将当前行写入对应的拆分文件
writer.writerow(row)
# 关闭所有拆分文件
for f in split_files.values():
f.close()
```
在上面的代码中,我们首先使用csv.reader读取csv文件,并使用next函数跳过表头。然后,我们遍历每一行数据,并获取第二列的值。如果拆分文件不存在,则创建一个新的文件,并将表头写入文件。接着,我们将当前行写入对应的拆分文件。最后,我们使用close函数关闭所有拆分文件。
你可以根据需要修改代码,例如根据其他列来拆分文件,或者使用csv.DictReader来读取csv文件。
阅读全文