Найти пересечение двух вложенных списков? - PullRequest
Новая
       16

Найти пересечение двух вложенных списков?

464 голосов
/ 13 марта 2009

Я знаю, как получить пересечение двух плоских списков: 

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]

или 

def intersect(a, b):
    return list(set(a) & set(b))

print intersect(b1, b2)

Но когда мне нужно найти пересечение для вложенных списков, тогда начинаются мои проблемы: 

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

В конце я хотел бы получить: 

c3 = [[13,32],[7,13,28],[1,6]]

Не могли бы вы, ребята, помочь мне с этим?

Относящиеся

Ответы [ 18 ]

3 голосов
/ 01 июля 2012

Не знаю, опоздал ли я на ваш вопрос. После прочтения вашего вопроса я придумал функцию intersect (), которая может работать как со списком, так и с вложенным списком. Я использовал рекурсию, чтобы определить эту функцию, она очень интуитивно понятна. Надеюсь, это то, что вы ищете: 

def intersect(a, b):
    result=[]
    for i in b:
        if isinstance(i,list):
            result.append(intersect(a,i))
        else:
            if i in a:
                 result.append(i)
    return result

Пример: 

>>> 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]]
>>> print intersect(c1,c2)
[[13, 32], [7, 13, 28], [1, 6]]

>>> b1 = [1,2,3,4,5,9,11,15]
>>> b2 = [4,5,6,7,8]
>>> print intersect(b1,b2)
[4, 5]
3 голосов
/ 13 марта 2009

Считаете ли вы, что [1,2] пересекается с [1, [2]]? То есть вас интересуют только числа или структура списка?

Если только цифры, выясните, как «сгладить» списки, затем используйте метод set().

1 голос
/ 18 мая 2017

Чтобы определить пересечение, которое правильно учитывает количество элементов, используйте Counter: 

from collections import Counter

>>> c1 = [1, 2, 2, 3, 4, 4, 4]
>>> c2 = [1, 2, 4, 4, 4, 4, 5]
>>> list((Counter(c1) & Counter(c2)).elements())
[1, 2, 4, 4, 4]
1 голос
/ 29 ноября 2016

Я также искал способ сделать это, и в итоге все закончилось так: 

def compareLists(a,b):
    removed = [x for x in a if x not in b]
    added = [x for x in b if x not in a]
    overlap = [x for x in a if x in b]
    return [removed,added,overlap]
0 голосов
/ 21 мая 2018 
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]]
c3 = [list(set(i) & set(c1)) for i in c2]
c3
[[32, 13], [28, 13, 7], [1, 6]]

Для меня это очень элегантный и быстрый способ к нему:)

0 голосов
/ 24 августа 2017 
# Problem:  Given c1 and c2:
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]]
# how do you get c3 to be [[13, 32], [7, 13, 28], [1, 6]] ?

Вот один из способов установки c3, который не включает наборы: 

c3 = []
for sublist in c2:
    c3.append([val for val in c1 if val in sublist])

Но если вы предпочитаете использовать только одну строку, вы можете сделать это: 

c3 = [[val for val in c1 if val in sublist]  for sublist in c2]

Это понимание списков внутри понимания списков, что немного необычно, но я думаю, вам не должно быть особых проблем с этим.

0 голосов
/ 21 ноября 2014 
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]]

c3 = [list(set(c2[i]).intersection(set(c1))) for i in xrange(len(c2))]

c3
->[[32, 13], [28, 13, 7], [1, 6]]
0 голосов
/ 05 мая 2015

Мы можем использовать для этого методы set: 

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

   result = [] 
   for li in c2:
       res = set(li) & set(c1)
       result.append(list(res))

   print result
Страница:
...