Zip nested list with list in Python

Question

Python newbie here. I have four lists, three of which are nested lists and one that isn't. I'm searching for a way to zip the nested lists with the list such that the zip function compares each nested list item with the corresponding item in the main list.

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]]]

I tried something like this:

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

but I'm getting the error message "too many values to unpack". Any suggestions?

It's not really clear how [False, True, False] relates to [2, 1, 3] — Open AI - Opting Out
– Open AI - Opting Out, Commented Jan 24, 2016 at 20:52
Your example lists are missing some commas. And True and False should be upper case. — Stefan Pochmann
– Stefan Pochmann, Commented Jan 24, 2016 at 22:31

alecxe · Accepted Answer · 2016-01-24 20:51:48Z

2

The idea is to zip() the main list with the already zipped a, b and c lists and make a nested 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]]]

answered Jan 24, 2016 at 20:51

alecxe

476k127 gold badges1.1k silver badges1.2k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Zip nested list with list in Python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related