更新时间:2023-12-03 21:22:22
您可以简单地使用字典,其中颜色是键,值是行.如果颜色已在字典中,请忽略该颜色,否则将其添加并将该行写入新的csv文件中.
You can simply use a dictionary where the color is the key and the value is the row. Ignore the color if it is already in the dictionary, otherwise add it and write the row to a new csv file.
import csv
file_in = 'input_file.csv'
file_out = 'output_file.csv'
with open(file_in, 'rb') as fin, open(file_out, 'wb') as fout:
reader = csv.reader(fin)
writer = csv.writer(fout)
d = {}
for row in reader:
color = row[0]
if color not in d:
d[color] = row
writer.writerow(row)
result = d.values()
result
# Output:
# [['blue', '9', 'center'],
# ['pink', '2677', 'left'],
# ['purple', '48', 'left'],
# ['yellow', '3222', 'right'],
# ['black', '123', 'left'],
# ['green', '3', 'center'],
# ['white', '68', 'right'],
# ['red', '75', 'right']]
以及csv文件的输出:
And the output of the csv file:
!cat output_file.csv
# Output:
# red,75,right
# green,3,center
# yellow,3222,right
# blue,9,center
# black,123,left
# white,68,right
# purple,48,left
# pink,2677,left