我有一个列表A:
A = [ [0, 42, 2.6],
[1, 20, 5.6],
[2, 67, 3.5],
[3, 12, 3.2],
[4, 34, 1.1],
[5, 74, 4.7],
[6, 29, 2.9] ]
我想从中提取一个子列表 B，其中仅包含那些第二列低于 30 的列表(即 B = [[1, 20, 5.6] ,[3, 12, 3.2],[6, 29, 2.9]])
当然，我可以将它转换为 numpy 并使用 np.where。但是我想知道是否有一种方法可以通过使用列表来做到这一点。我尝试了列表理解，但我不是很擅长:
B = [x for x in len(A) if x[1] <= 30]
但它确实不起作用。有什么建议吗？提前致谢。
请您参考如下方法:
试试这个:
B = [x for x in A if x[1] <= 30]
