Find intersection of two nested lists?

前端 未结 20 1132
星月不相逢
星月不相逢 2020-11-22 04:16

I know how to get an intersection of two flat lists:

b1 = [1,2,3,4,5,9,11,15]
b2 = [4,5,6,7,8]
b3 = [val for val in b1 if val in b2]

or

<
20条回答
  •  长发绾君心
    2020-11-22 05:12

    Pure list comprehension version

    >>> c1 = [1, 6, 7, 10, 13, 28, 32, 41, 58, 63]
    >>> c2 = [[13, 17, 18, 21, 32], [7, 11, 13, 14, 28], [1, 5, 6, 8, 15, 16]]
    >>> c1set = frozenset(c1)
    

    Flatten variant:

    >>> [n for lst in c2 for n in lst if n in c1set]
    [13, 32, 7, 13, 28, 1, 6]
    

    Nested variant:

    >>> [[n for n in lst if n in c1set] for lst in c2]
    [[13, 32], [7, 13, 28], [1, 6]]
    

提交回复
热议问题