python - 输出带有列表理解的嵌套列表

Question

我有以下列表:

 a = [[0,11], [1,12], [3,14], [5,16],[7,18]]

我想创建一个新的列表 b 来满足所有元素

 a[:,0] > 3

并在所选嵌套列表的 a 中添加索引。所以 b 应该看起来像:

b = [[3, 5, 16],[4, 7,18]]

与:

b = [points, points in a if points[0]>3]

将输出带有索引的 b = [[5, 16],[7,18]]，并且

b = [[index, points], index, points in enumerate(a) if points[0]>3]

显示错误。如何使用列表理解生成 b？

Answer 1

您可以使用一次性变量的元组来保存嵌套列表项:

>>> [[i,j,k] for i,(j,k) in enumerate(a) if j>3]
[[3, 5, 16], [4, 7, 18]]

或者作为包含更多项目的列表的更全面的方法，您可以使用以下列表理解:

>>> [[i] + points for i, points in enumerate(a) if points[0]>3]
[[3, 5, 16], [4, 7, 18]]

使用解包分配的 Python 3 版本:

>>> [[i, first] + rest for i,(first, *rest) in enumerate(a) if first>3]
[[3, 5, 16], [4, 7, 18]]