python对csv文件中的某一列数据去重,并将去重后的该列结果存入新的csv文件中
时间: 2023-05-24 19:07:36 浏览: 313
以下是Python代码示例,可以实现对CSV文件中的某一列数据去重,并将去重后的结果存入新的CSV文件中:
```python
import csv
# 读取原始CSV文件
filename = "original.csv"
with open(filename, "r") as file:
reader = csv.reader(file)
data = list(reader)
# 获取需要去重的列的索引
column_index = 2 # 假设需要去重的列是第三列(索引从0开始计数)
# 将需要去重的列的数据存入集合中进行去重
unique_items = set()
for row in data:
unique_items.add(row[column_index])
# 将去重后的结果存入新的CSV文件中
new_filename = "unique.csv"
with open(new_filename, "w", newline="") as file:
writer = csv.writer(file)
for item in unique_items:
writer.writerow([item])
```
在上述代码中,首先使用Python的csv模块读取原始CSV文件,将其转化为一个二维数组。然后通过设置column_index变量,确定需要去重的列的索引。接着,使用Python的set数据类型来存放需要去重的列的数据,并调用其自带的去重功能。最后,将去重后的结果存入新的CSV文件中,新文件名为unique.csv。在写入新的CSV文件时,需要将每个唯一的数据项分别写入一行,因此需要使用csv.writer中的writerow函数。