是否要返回多维列表中的特定计数？

Question

我有一个多维列表，我想返回一定的长度，但是知道要写什么代码。例如，我下面有一个列表，我想返回列表中长度为4或更大的所有数字。

A=[[2,4,6,8],[2,12,20],[34,35,37,38],[4,8,9,20,21],[5,7,9,12]]

我希望我的结果是

B=[[2,4,6,8],[34,35,37,38],[5,7,9,12]]

Answer 1

根据您期望的输出，假设您实际上想要的子列表的长度为4，而不是4或更长，您可以使用列表推导，过滤子列表的长度是否为4个元素：

B = [l for l in A if len(l) == 4]

输出：

[[2, 4, 6, 8], [34, 35, 37, 38], [5, 7, 9, 12]]

如果您想要4个或更多，请将==更改为>=：

B = [l for l in A if len(l) >= 4]

输出：

[[2, 4, 6, 8], [34, 35, 37, 38], [4, 8, 9, 20, 21], [5, 7, 9, 12]]