python - 使用 Python 从行列表中提取列

Question

我有以下数据集，它是存储为嵌套列表的一系列行:

[['John', '35', 'UK'],
['Emma', '43', 'UK'],
['Lucy', '25', 'AU']]

(行总是相同的长度)

我需要将 'UK', 'AU' 作为可迭代对象返回(与顺序无关)。

是否有一个单行代码返回第三列中包含的唯一值，并且比这更简单？

set(list(map(list, zip(*l)))[2])

(引用:Transpose list of lists)

Answer 1

更改您自己的代码:

Python 3.x:

set(list(zip(*l)[2]))

python 2.x:

set(zip(*l)[2])

演示:

l=[['John', '35', 'UK'],['Emma', '43', 'UK'],['Lucy', '25', 'AU']]
set(list(zip(*l)[2]))
{'AU', 'UK'}