如何使用python获取csv文件中第一列仅出现一次的值

Question

我是python新手，所以我尝试读取包含700行的csv，其中包含一个标题，并获得一个包含第一个csv列的唯一值的列表。你知道吗

示例CSV：

SKU;PRICE;SUPPLIER
X100;100;ABC
X100;120;ADD
X101;110;ABV
X102;100;ABC
X102;105;ABV
X100;119;ABG

我用了这个例子

How to create a list in Python with the unique values of a CSV file?

所以我做了以下工作：

import csv
mainlist=[]
with open('final_csv.csv', 'r', encoding='utf-8') as csvf:
    rows = csv.reader(csvf, delimiter=";")
    for row in rows:
        if row[0] not in rows:
            mainlist.append(row[0])
print(mainlist)

我注意到在调试中，行是1行而不是700行，我只得到

['SKU']字段我做错了什么？你知道吗

谢谢

Answer 1

如果需要第一列的唯一值，可以修改代码以使用set而不是list。可能是这样的：

import collections
import csv
filename = 'final_csv.csv'

sku_list = []
with open(filename, 'r', encoding='utf-8') as f:
    csv_reader = csv.reader(f, delimiter=";")

    for i, row in enumerate(csv_reader):
        if i == 0:
            # skip the header
            continue

        try:
            sku = row[0]
            sku_list.append(sku)
        except IndexError:
            pass

print('All SKUs:')
print(sku_list)

sku_set = set(sku_list)
print('SKUs after removing duplicates:')
print(sku_set)

c = collections.Counter(sku_list)
sku_list_2 = [k for k, v in c.items() if v == 1]
print('SKUs that appear only once:')
print(sku_list_2)

with open('output.csv', 'w') as f:
    for sku in sorted(sku_set):
        f.write('{}\n'.format(sku))