如何从元组列表中提取 n 个元素

n = 1 # N. . .
[x[n] for x in elements]

我知道可以用 FOR 来完成，但是我想知道是否还有其他的方法

还有另一种方法。你也可以用 地图和 项目获得者来做:

>>> from operator import itemgetter
>>> map(itemgetter(1), elements)

尽管如此，它仍然在内部执行一个循环，并且比列表内涵略慢:

setup = 'elements = [(1,1,1) for _ in range(100000)];from operator import itemgetter'
method1 = '[x[1] for x in elements]'
method2 = 'map(itemgetter(1), elements)'


import timeit
t = timeit.Timer(method1, setup)
print('Method 1: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup)
print('Method 2: ' + str(t.timeit(100)))

结果:

Method 1: 1.25699996948
Method 2: 1.46600008011

如果需要在列表上迭代，那么使用 for就可以了。

这也行得通:

zip(*elements)[1]

(我主要是发布这个，以证明自己，我已经吸引了 zip...)

看看它的实际效果:

>>> help(zip)

模块 天生的中内置函数 zip 的帮助:

拉链(...)

Zip (seq1[ ，seq2[ ... ])-> [(seq1[0] ，seq2[0] ...) ，(...)]

返回一个元组列表，其中每个元组包含第 i 个元素 返回的列表被截断 以最短参数序列的长度为单位。

>>> elements = [(1,1,1),(2,3,7),(3,5,10)]
>>> zip(*elements)
[(1, 2, 3), (1, 3, 5), (1, 7, 10)]
>>> zip(*elements)[1]
(1, 3, 5)
>>>

我今天学到了一件很棒的事情: 在参数中使用 *list为函数创建一个参数列表..。

注意 : 在 Python 3中，zip返回一个迭代器，因此使用 list(zip(*elements))返回一个元组列表。

当我在寻找最快的方法来获取2元组列表中的第二个元素时发现了这个问题。这不是我想要的，但是运行了与第3个方法相同的测试并测试了 zip 方法

setup = 'elements = [(1,1) for _ in range(100000)];from operator import itemgetter'
method1 = '[x[1] for x in elements]'
method2 = 'map(itemgetter(1), elements)'
method3 = 'dict(elements).values()'
method4 = 'zip(*elements)[1]'


import timeit
t = timeit.Timer(method1, setup)
print('Method 1: ' + str(t.timeit(100)))
t = timeit.Timer(method2, setup)
print('Method 2: ' + str(t.timeit(100)))
t = timeit.Timer(method3, setup)
print('Method 3: ' + str(t.timeit(100)))
t = timeit.Timer(method4, setup)
print('Method 4: ' + str(t.timeit(100)))


Method 1: 0.618785858154
Method 2: 0.711684942245
Method 3: 0.298138141632
Method 4: 1.32586884499

所以如果你有一个2元组对只是转换成一个 dict 并获取值，速度会快两倍以上。

map (lambda x:(x[1]),elements)

从2元组列表中提取第二个元素的 Python 3.6计时。

此外，还增加了 numpy数组方法，这种方法读起来更简单(但可以说比列表内涵更简单)。

from operator import itemgetter
elements = [(1,1) for _ in range(100000)]


%timeit second = [x[1] for x in elements]
%timeit second = list(map(itemgetter(1), elements))
%timeit second = dict(elements).values()
%timeit second = list(zip(*elements))[1]
%timeit second = np.array(elements)[:,1]

以及时机:

list comprehension:  4.73 ms ± 206 µs per loop
list(map):           5.3 ms ± 167 µs per loop
dict:                2.25 ms ± 103 µs per loop
list(zip)            5.2 ms ± 252 µs per loop
numpy array:        28.7 ms ± 1.88 ms per loop

注意，map()和 zip()不再返回列表，因此需要进行显式转换。

使用 islice和 chain.from_iterable:

>>> from itertools import chain, islice
>>> elements = [(1,1,1),(2,3,7),(3,5,10)]
>>> list(chain.from_iterable(islice(item, 1, 2) for item in elements))
[1, 3, 5]

当你需要不止一个元素时，这会很有用:

>>> elements = [(0, 1, 2, 3, 4, 5),
(10, 11, 12, 13, 14, 15),
(20, 21, 22, 23, 24, 25)]
>>> list(chain.from_iterable(islice(tuple_, 2, 5) for tuple_ in elements))
[2, 3, 4, 12, 13, 14, 22, 23, 24]

我喜欢@daren 的回答，但是它给出了一个错误，即 zip不可订阅，因为您无法切分 zip迭代器。相反，您需要将 zip添加到 list中，以根据索引提取所需的元素。

elements = [(1,1,1),(2,3,7),(3,5,10)]
slices = list(zip(*elements))[1]

产出:

(1, 3, 5)