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

Question

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

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

我想提取具有相同 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

好吧，与您之前的问题相比，您添加了额外的列表深度

所以，这个列表理解有效：

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