使用python的csv文件中列中的唯一元素

Question

我有一个分号分隔的 csv 文件，格式如下：

indx1; string1; char1; entry1 
indx2; string1; char2; entry2 
indx3; string2; char2; entry3 
indx4; string1; char1; entry4 
indx5; string3; char2; entry5 

我想以列表的形式（不使用 pandas 或 numpy）获取此文件第一列和第二列的唯一条目。特别是这些是我想要的列表：

[string1, string2, string3] 
[char1, char2]

顺序无所谓，希望操作快点。

目前，我正在使用命令读取文件（比如“data.csv”）

with open('data.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=';')

我正在使用 python 2.7。实现我想要的功能的最快方法是什么？我将不胜感激。

Answer 1

您可以使用sets 来跟踪所需列中已经看到的值。既然您说顺序无关紧要，您可以在处理完所有行后将集合转换为列表：

import csv

col1, col2 = set(), set()

with open('data.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=';', skipinitialspace=True)        
    for row in csv_reader:
        col1.add(row[1])
        col2.add(row[2])

print list(col1), list(col2)  # ['string1', 'string3', 'string2'] ['char2', 'char1']

Answer 2

这应该可行。您可以将其用作基准。

myDict1 = {}
myDict2 = {}
with open('data.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=';')
    for row in csv_reader:
        myDict1[row[1]] = 0
        myDict2[row[2]] = 0

x = myDict1.keys() 
y = myDict2.keys()