Question

Python新手在这里。我有四个列表，其中三个是嵌套列表，另一个是不列表。我正在搜索使用列表压缩嵌套列表的方法，以便zip函数将每个嵌套列表项与主列表中的相应项进行比较。

main = [1,3]
a = [[1,2,3][4,5,6]]
b = [[0,1,2][3,4,5]]
c = [[2,3,4][5,6,7]]

>>>[[[True, False, False],[False,True,False],[False,False,False]],
[[False,False,False],[True,False,False],[False,False,False]]]

我试过这样的事情：

abc = zip(a,b,c)
test = (x==y for x, y in zip(main,*abc)

但是我收到了错误消息＆＃34;太多的值无法打开＆＃34;。有什么建议吗？

Answer 1

我们的想法是zip() main列表，其中包含已经压缩的a，b和c列表，并制作嵌套的list comprehension ：

>>> [[[item == x for item in l] for l in lists] 
     for x, lists in zip(main, zip(a, b, c))]
[[[True, False, False], [False, True, False], [False, False, False]], 
 [[False, False, False], [True, False, False], [False, False, False]]]

使用Python中的列表压缩嵌套列表

1 个答案: