python读取csv文件某一列并另存为csv文件
时间: 2023-11-20 16:53:05 浏览: 122
要读取csv文件某一列并另存为csv文件,可以使用Python的csv模块中的DictReader和DictWriter方法。具体步骤如下:
1. 使用DictReader方法读取原始csv文件,将其转换为字典格式。
2. 通过字典的键值对,获取需要的列数据。
3. 使用DictWriter方法将所需列数据写入新的csv文件中。
下面是一个示例代码:
```
import csv
# 读取原始csv文件
with open('input.csv', 'r') as input_file:
reader = csv.DictReader(input_file)
# 获取需要的列数据
column_data = [row['column_name'] for row in reader]
# 将所需列数据写入新的csv文件中
with open('output.csv', 'w', newline='') as output_file:
writer = csv.DictWriter(output_file, fieldnames=['column_name'])
writer.writeheader()
for data in column_data:
writer.writerow({'column_name': data})
```
其中,'column_name'需要替换为实际需要读取的列名。另外,需要注意的是,如果原始csv文件中存在空行,需要在打开文件时指定newline=''参数,否则会出现空行。
阅读全文