Question

如何从下面列出的列表中提取特定元组：

[[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]]

我想用SAME x轴提取元组，如：

[(1,4), (1,3), (1,2)]

而（5,4）应该被丢弃。请帮帮我感谢

Answer 1

要展平列表，您应该始终使用itertools.chain。在这种情况下，您必须应用嵌套的chain来创建元组的平面列表

>>> l=[[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]]
>>> [e for e in chain(*chain(*l)) if e[0] == 1]
[(1, 4), (1, 3), (1, 2)]

Answer 2

与您的previous question

相比，您添加了更多深度列表

所以，这个列表理解起作用：

>>> l=[[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]]
>>> def flat1(l): return [i for sub in l for i in sub]
...
>>> [t for t in flat1(flat1(l)) if t[0]==1]
[(1, 4), (1, 3), (1, 2)]

如果与itertools.chain(*l)的速度相同（稍快），则列表列表列表的列表理解方法。

如果你想使用itertools来提高速度，你应该使用chain.from_iterable(l)代替。

以下是时间：

import timeit
n=1000000

c1='''
def f1(l):
    return [e for e in itertools.chain(*itertools.chain(*l)) if e[0] == 1]

l1=f1([[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]])
'''    

c2='''
def f2(l):
    def flat1(l): return [i for sub in l for i in sub]
    return [t for t in flat1(flat1(l)) if t[0]==1]

l1=f2([[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]])
'''

c3='''
def f1(l):
    flat1=itertools.chain.from_iterable
    return [e for e in flat1(flat1(l)) if e[0] == 1]

l1=f1([[[(1, 4)]], [[(1, 3)], [(5, 4)]], [[(1, 2)]]])
'''              

t1=timeit.timeit(stmt=c1,setup='import itertools',number=n)
t2=timeit.timeit(stmt=c2,number=n)
t3=timeit.timeit(stmt=c3,setup='import itertools',number=n)

print '          chain(*l):',t1,'seconds'
print ' list comprehension:',t2,'seconds'
print 'chain.from_iterable:',t3,'seconds'

打印：

          chain(*l): 4.32919406891 seconds
 list comprehension: 4.32601380348 seconds
chain.from_iterable: 3.14966917038 seconds

如何从python中的列表中提取特定元组？

2 个答案: