通过在python中比较两个嵌套列表来生成0和1的嵌套列表

时间:2019-08-03 15:37:29

标签: python list nested

我有两个嵌套列表,如下所示:

list_x = [[21, 58, 68, 220, 266, 386, 408, 505, 518, 579], 
          [283, 286, 291, 321, 323, 372, 378, 484, 586, 629]]

list_y = [[21, 220, 386, 505, 518], [286, 291, 321, 323, 372]]

我想比较上述嵌套列表中具有相同索引位置的元素,这意味着list_x[0]应该与list_y[0]比较,依此类推。

我想生成第三个(嵌套的)列表,以便对于list_x[0]中的每个数字,如果该数字也位于list_y[0]中,则生成一个,如果不匹配,则生成一个零生成。对list_x[1]list_y[1]执行相同的过程。

我的嵌套输出列表中每个子列表的长度应为10(即较长的子列表的长度,一个为匹配项,一个为零,否则为零)。所有子列表均按升序排序。

一些值得共享的其他信息是list_y[0]list_y[1]分别是list_x[0]list_x[1]的子集。

因此,我要搜索的输出列表如下:

out = [[1,0,0,1,0,1,0,1,1,0], [0,1,1,1,1,1,0,0,0,0]]

我尝试了以下代码,但又得到了10个额外的零

list_x = [y for x in list_x for y in x] #to flatten list_x

result = []
for y in list_y:
    sublist = []
    for x in list_x:
        if x in y:
            sublist.append(1)
        else: 
            sublist.append(0)
    result.append(sublist)

上面的代码为我提供了以下内容:

result = [[1, 0, 0, 1, 0, 1, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
          [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]]

谢谢您的帮助!

3 个答案:

答案 0 :(得分:1)

我们可以使用zip并发地遍历子列表,然后执行in检查,例如:

[[int(x in suby) for x in subx] for subx, suby in zip(list_x, map(set, list_y))]

然后产生:

>>> [[int(x in suby) for x in subx] for subx, suby in zip(list_x, list_y)]
[[1, 0, 0, 1, 0, 1, 0, 1, 1, 0], [0, 1, 1, 1, 1, 1, 0, 0, 0, 0]]

map(set, list_y)用于映射集中的list_y的所有子列表,因为对集合的查找通常会在 O(1)中运行,而查找列表中的 O(n)

答案 1 :(得分:0)

托马斯,欢迎您!

尝试一下:

#!/usr/bin/env python2

list_x = [[21, 58, 68, 220, 266, 386, 408, 505, 518, 579],
          [283, 286, 291, 321, 323, 372, 378, 484, 586, 629]]

list_y = [[21, 220, 386, 505, 518], [286, 291, 321, 323, 372]]

answer=[]
for ( index, inner_list ) in enumerate( list_x ):
  answer.append([])

  for ( inner_index, inner_value ) in enumerate(inner_list):
    answer[index].append(0)
    if inner_value in list_y[ index ]:
      answer[index][inner_index] = 1
print answer

答案 2 :(得分:0)

尝试一下...您将获得输出

    list_x = [[21, 58, 68, 220, 266, 386, 408, 505, 518, 579], 
      [283, 286, 291, 321, 323, 372, 378, 484, 586, 629]]

    list_y = [[21, 220, 386, 505, 518], [286, 291, 321, 323, 372]]
    result =[]
    for ind,lst in enumerate(list_x):
        sublist  =[]
        for ele in lst:
            if ele in list_y[ind]:
                sublist .append(1)
            else:
                sublist .append(0)
         result.append(sublist )
    print(result)

输出

    [[1,0,0,1,0,1,0,1,1,0], [0,1,1,1,1,1,0,0,0,0]]