选择列表中一定长度的列表

时间:2019-03-13 23:59:45

标签: python python-3.x nested-lists

这就是我所拥有的:

original_list=[[1,2],[1,2,3],[1,2],[1,2,3,4],[1,2,3]]

我想在列表中找到长度为2、3、4等的列表:

length2_list=[[1,2],[1,2]]

length3_list=[[1,2,3],[1,2,3]]

length4_list=[[1,2,3,4]]

我该怎么办?

3 个答案:

答案 0 :(得分:2)

列表理解

#don't call lists "list" as a variable 
l=[[1,2],[1,2,3],[1,2],[1,2,3,4],[1,2,3]]

#for len 2
l_2=[x for x in l if len(x)==2]

#for len 3
l_3=[x for x in l if len(x)==3]

答案 1 :(得分:1)

使用Len()函数很简单,它可以告诉您列表的长度

origal_list=[[1,2],[1,2,3],[1,2],[1,2,3,4],[1,2,3]]

length2_list=[]

length3_list=[]

length4_list=[]


for lst in origal_list:
    if len(lst) == 2:
        length2_list.append(lst)
    if len(lst) == 3:
        length3_list.append(lst)
    if len(lst) == 4:
        length4_list.append(lst)

print(length2_list)
print(length3_list)
print(length4_list)

结果:

[[1, 2], [1, 2]]
[[1, 2, 3], [1, 2, 3]]
[[1, 2, 3, 4]]

答案 2 :(得分:0)

尝试类似的方法,它将适用于n个长度为n的列表:

lists = {}
list = [[1, 2], [1, 2, 3], [1, 2], [1, 2, 3, 4], [1, 2, 3]]

for l in list:
    length = len(l)
    if lists.get(length) is None:
        lists[length] = []
    lists[length].append(l)

这将创建一个字典,该字典的长度将作为列表列表的关键字。然后,它将遍历原始列表,并将其附加到正确的密钥上。

它将输出如下内容:{2: [[1, 2], [1, 2]], 3: [[1, 2, 3], [1, 2, 3]], 4: [[1, 2, 3, 4]]}