python中CSV文件的所有列的唯一元素，而无需使用pandas

Question

我正在尝试获取CSV中所有列的唯一值。我正在获取所有列的列号并创建集合，并尝试浏览CSV数据并找到唯一的列。但是第二个循环仅执行一次。

decoded_file = data_file.read().decode('utf-8')
reader = csv.reader(decoded_file.splitlines(),
                            delimiter=',')
list_reader = list(reader)
data = iter(list_reader)
next(data) #skipping the header
col_number = len(next(data))
col_sets = [set() for i in range(col_number)]

for col in range(col_number):
   for new_row in data:
       col_sets[col].add(new_row[col])
   print(col_sets[col])

我需要为每列获取所有唯一值，并将其添加到col_sets以访问它。最好的方法是什么？

Answer 1

一切都很好，但是您应该只更改迭代顺序。

for new_row in data:
    for col in range(col_number):
        col_sets[col].add(new_row[col])
print(col_sets)