python - 在嵌套列表中获取特定索引的不同值

Question

我有一个包含大约 100 万条记录的嵌套列表，例如:

l = [['a', 'b', 'c', ...], ['d', 'b', 'e', ...], ['f', 'z', 'g', ...],...]

我想在第二个索引上获取内部列表的不同值，以便我的结果列表如下:

resultant = ['b', 'z', ...]

我试过嵌套循环，但速度不快，我们将不胜感激!

Answer 1

因为你想要独特的项目，你可以使用 collections.OrderedDict.fromkeys() 来保持顺序和独特的项目(因为使用哈希表来回键)并使用 zip( ) 获取第二项。

from collections import OrderedDict

list(OrderedDict.fromkeys(zip(my_lists)[2]))

在 python 3.x 中，因为 zip() 返回一个迭代器，你可以这样做:

colls = zip(my_lists)
next(colls)
list(OrderedDict.fromkeys(next(colls)))

或者在 dict.formkeys() 中使用生成器表达式:

list(OrderedDict.fromkeys(i[1] for i in my_lists))

演示:

>>> lst = [['a', 'b', 'c'], ['d', 'b', 'e'], ['f', 'z', 'g']]
>>> 
>>> list(OrderedDict().fromkeys(sub[1] for sub in lst))
['b', 'z']