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

Question

我试图从元组列表中获取第n个元素.

我有类似的东西:

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

我希望只将每个元组的第二个元素提取到一个列表中:

seconds = [1, 3, 5]

我知道它可以通过for循环来完成,但我想知道是否还有另一种方法,因为我有数千个元组.

Answer 1

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

Answer 2

我知道可以用FOR来完成,但我想知道是否还有另一种方法

还有另一种方式.你也可以用map和itemgetter来做:

>>> 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

如果你需要迭代列表然后使用a for就可以了.

Answer 3

这也有效:

zip(*elements)[1]

(我主要发布这个,向我自己证明我已经开始了zip......)

看到它的实际效果:

>>> help(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在参数中使用为函数创建参数列表......

注意:在Python3中,zip返回一个迭代器,因此list(zip(*elements))用于返回元组列表.

Answer 4

找到这个,因为我正在寻找哪种方式最快拉出2元组列表的第二个元素.不是我想要的,而是使用第三种方法运行相同的测试,再加上测试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并获取值,则速度提高两倍.

Answer 5

定时为Python的3.6，用于从2元组列表中提取第二元件。

此外，添加了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()不再返回列表，因此是显式转换。