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

Question

我想从元组列表中获取第n个元素。

我有类似：

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

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

seconds = [1, 3, 5] 

我知道，它可以与for循环进行，但我想知道是否有另一种方式，因为我有成千上万的元组。

Answer 1

[x[1] 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 

如果你需要迭代一个列表，那么使用for就可以了。

Answer 3

这也适用于：

zip(*elements)[1] 

（我主要是发布此，为了证明自己，我已经groked 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在参数来创建一个函数的参数列表...

Answer 4

发现这是我在寻找它的方式是最快的拉第二个元素的2元组列表。不是我想要的，但跑了相同的测试，如图与3方法以及测试方法拉链

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对，只是转换到一个字典，并采取值。

Answer 5

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

Answer 6

计时为的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()再也不会返回一个列表，因此显式转换。