Python：无论如何使用map来获取元组的第一个元素

Question

我有一个元组的元组，我想将每个元组中的第一个值放入一个集合中。我认为使用 map() 是一个很好的方法，唯一的事情是我找不到一种简单的方法来访问元组中的第一个元素。例如，我有元组 ((1,), (3,))。我想做类似 set(map([0], ((1,), (3,)))) 的事情（其中 [0] 正在访问第零个元素）得到一个包含 1 和 3 的集合。我能想到的唯一方法是定义一个函数：def first(t): return t[0]。无论如何，是否可以在一行中完成此操作而无需声明该函数？

Answer 1

使用列表理解：

data = ((1,), (3,))
print([x[0] for x in data])

Answer 2

使用 operator.itemgetter

from operator import itemgetter
map(itemgetter(0), ((1,), (3,)))

：列表理解通常更具可读性， itemgetter 最接近您的要求。它也更快一点：

>>> from timeit import timeit
>>> setup = 'from operator import itemgetter; lst=( ("a",), ("b",), (1,), (2,))'
>>> timeit('map(itemgetter(0), lst)', setup=setup)
0.13061050399846863
>>> timeit('[i[0] for i in lst]', setup=setup)
0.20302422800159547

Answer 3

my_set = {x[0] for x in TUPLES}

Answer 4

还有另一种获取方式：

set(x for x, in data)

Answer 5

Python 支持使用 lambda 创建匿名函数a> 关键字。这允许您使用函数而无需正式定义它。给定您的示例，您将像这样使用 lambda：

data = ((1,), (3,))
set(map(lambda x: x[0], data))

这相当于：

def f(x):
    return x[0]

set(map(f, data))

但正如其他人所说，列表推导式优于使用映射。

Answer 6

您可以在 Python 2.7 和 3.x 中使用集合理解：

>>> t = ((1,), (3,))
>>> s = {x[0] for x in t}
>>> s
set([1, 3])

或在 Python < 2.7:

>>> s = set([x[0] for x in t])
>>> s
set([1, 3])

Answer 7

data = ((1,), (3,))
s = set(zip(*data)[0])

如果元组中有更多项目，您可以使用 izip 和 islice 节省一些内存和时间。

Answer 8

和@Winston 一起去。列表理解非常棒。如果您确实想要使用映射，请按照之前的建议使用 lambda，或者逻辑上等效的...

from operator import itemgetter
data = ((1,), (3,))
map(itemgetter(0), data)

这仅供参考；您应该使用列表比较