找到所有元素和指标大于阈值的列表清单

Question

我有这样一个列表的列表：

j=[[1,2,3],[4,5,6],[7,8,9,10],[11,12,13,14,15]] 

，我想从超过某一阈值列表清单中获得所有元素的列表。我知道什么时候有一个列表有语法：

k=[1,2,3,4,5,6,7] 
k2=[i for i in k if i>5] 
k2=[6,7] 

当列表作为输入列表时，是否有类似的方式？ 是否还有可能使元素的索引大于该阈值？

Answer 1

这是你的意思吗？

In [3]: l1 = [[1,2,3],[4,5,6],[7,8,9,10],[11,12,13,14,15]] 
In [4]: [e for l2 in l1 for e in l2 if e>5] 
Out[4]: [6, 7, 8, 9, 10, 11, 12, 13, 14, 15] 

为了得到（内列出的）的指数，你可以使用enumerate：

In [5]: [(i2, e) for l2 in l1 for i2, e in enumerate(l2) if e>5] 

Out[5]: 
[(2, 6), 
(0, 7), 
(1, 8), 
(2, 9), 
(3, 10), 
(0, 11), 
(1, 12), 
(2, 13), 
(3, 14), 
(4, 15)] 

如果你想在一个单独的列表索引（UN）邮编：

In [6]: list(zip(*[(i2, e) for l2 in l1 for i2, e in enumerate(l2) if e>5])) 

Out[6]: [(2, 0, 1, 2, 3, 0, 1, 2, 3, 4), (6, 7, 8, 9, 10, 11, 12, 13, 14, 15)] 

[如果您使用的是Python 2，则不需要拨打list]。类似的方法会给你索引到两个外部和内部列表：

In [7]: [((i1,i2), e) for (i1,l2) in enumerate(l1) for i2, e in enumerate(l2) if e>5] 
Out[7]: 
[((1, 2), 6), 
((2, 0), 7), 
((2, 1), 8), 
((2, 2), 9), 
((2, 3), 10), 
((3, 0), 11), 
((3, 1), 12), 
((3, 2), 13), 
((3, 3), 14), 
((3, 4), 15)] 

，或者解压缩：

In [8]: idx, items = list(zip(*[((i1,i2), e) for (i1,l2) in enumerate(l1) 
            for i2, e in enumerate(l2) if e>5])) 

In [8]: idx 
Out[8]: 
((1, 2), 
(2, 0), 
(2, 1), 
(2, 2), 
(2, 3), 
(3, 0), 
(3, 1), 
(3, 2), 
(3, 3), 
(3, 4)) 

In [9]: items 
Out[9]: (6, 7, 8, 9, 10, 11, 12, 13, 14, 15)