我正在尝试从元组列表中获取第n个元素。
我有类似的东西:
elements = [(1,1,1),(2,3,7),(3,5,10)]
我希望只将每个元组的第二个元素提取到列表中:
seconds = [1, 3, 5]
我知道可以通过for
循环来完成,但我想知道是否还有另外一种方法,因为我有数千个元组。
答案 0 :(得分:159)
n = 1 # N. . .
[x[n] for x in elements]
答案 1 :(得分:27)
我知道可以用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
就可以了。
答案 2 :(得分:25)
这也有效:
zip(*elements)[1]
(我主要是张贴这个,向我自己证明我已经试过zip
...)
看到它的实际效果:
>>> help(zip)
有关内置函数zip的帮助 builtin :
拉链(...)
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))
返回元组列表。
答案 3 :(得分:11)
发现这一点,因为我正在寻找哪种方式最快拉出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并获取值,那么速度超过两倍。
答案 4 :(得分:3)
map (lambda x:(x[1]),elements)
答案 5 :(得分:3)
用于 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()
不再返回列表,因此显式转换。
答案 6 :(得分:1)
>>> 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]