Видаліть усі елементи, що зустрічаються в одному списку, з іншого

Скажімо, у мене є два списки, l1і l2. Я хочу виконати l1 - l2, що повертає всі елементи l1не в l2.

Я можу придумати наївний цикл підходу до цього, але це буде справді неефективно. Який пітонічний та ефективний спосіб зробити це?

Як приклад, якщо у мене є l1 = [1,2,6,8] and l2 = [2,3,5,8], l1 - l2слід повернутися[1,6]

У Python є мовна функція під назвою Список Зрозуміння, яка ідеально підходить для того, щоб зробити цю річ надзвичайно легкою. Наступне твердження робить саме те, що ви хочете, і зберігає результат у l3:

l3 = [x for x in l1 if x not in l2]

l3буде містити [1, 6].

Одним із способів є використання наборів:

>>> set([1,2,6,8]) - set([2,3,5,8])
set([1, 6])

Як альтернативу, ви також можете використовувати filterз виразом лямбда, щоб отримати бажаний результат. Наприклад:

>>> l1 = [1,2,6,8]
>>> l2 = set([2,3,5,8])

#     v  `filter` returns the a iterator object. Here I'm type-casting 
#     v  it to `list` in order to display the resultant value
>>> list(filter(lambda x: x not in l2, l1))
[1, 6]

Порівняння продуктивності

Тут я порівнюю ефективність усіх відповідей, згаданих тут. Як і очікувалося, операція на set базі Арку є найшвидшою.

• Різниця набору Arkku - перше (0,124 Usec за цикл)

  mquadri$ python -m timeit -s "l1 = set([1,2,6,8]); l2 = set([2,3,5,8]);" "l1 - l2"
  10000000 loops, best of 3: 0.124 usec per loop

• Зрозуміння списку Даніеля Прийдена з setпошуком - Друге (0,302 Usec за цикл)

  mquadri$ python -m timeit -s "l1 = [1,2,6,8]; l2 = set([2,3,5,8]);" "[x for x in l1 if x not in l2]"
  1000000 loops, best of 3: 0.302 usec per loop

• Зрозуміння списку пончиків у звичайному списку - Третє (0,552 використання в циклі)

  mquadri$ python -m timeit -s "l1 = [1,2,6,8]; l2 = [2,3,5,8];" "[x for x in l1 if x not in l2]"
  1000000 loops, best of 3: 0.552 usec per loop

• Використанняfilter Мойнддіна Квадрі - четверте (0,972 використання в циклі)

  mquadri$ python -m timeit -s "l1 = [1,2,6,8]; l2 = set([2,3,5,8]);" "filter(lambda x: x not in l2, l1)"
  1000000 loops, best of 3: 0.972 usec per loop

• Комбінація Akshay Hazari reduce+filter - П'ята (3,97 Usec за цикл)

  mquadri$ python -m timeit "l1 = [1,2,6,8]; l2 = [2,3,5,8];" "reduce(lambda x,y : filter(lambda z: z!=y,x) ,l1,l2)"
  100000 loops, best of 3: 3.97 usec per loop

PS: set не підтримує порядок і видаляє повторювані елементи зі списку. Отже, не використовуйте різницю, якщо вам потрібно щось із цього.

Розширюючи відповідь Пончика та інші відповіді тут, ви можете отримати ще кращі результати, використовуючи розуміння генератора замість розуміння списку та використовуючи setструктуру даних (оскільки inоператор O (n) у списку, але O (1) на набір).

Ось ось функція, яка працювала б для вас:

def filter_list(full_list, excludes):
    s = set(excludes)
    return (x for x in full_list if x not in s)

У результаті вийде ітерабельний, який буде ліниво отримувати відфільтрований список. Якщо вам потрібен реальний об'єкт списку (наприклад, якщо вам потрібно зробити len()результат за результатом), ви можете легко створити такий список, як:

filtered_list = list(filter_list(full_list, excludes))

Використовуйте тип набору Python. Це було б найбільш піфонічно. :)

Крім того, оскільки він рідний, він повинен бути і найбільш оптимізованим методом.

Подивитися:

http://docs.python.org/library/stdtypes.html#set

http://docs.python.org/library/sets.htm (для старшого python)

# Using Python 2.7 set literal format.
# Otherwise, use: l1 = set([1,2,6,8])
#
l1 = {1,2,6,8}
l2 = {2,3,5,8}
l3 = l1 - l2

використовуйте Встановити розуміння {x для x в l2} або set (l2), щоб встановити, а потім використовуйте Список Поняття, щоб отримати список

l2set = set(l2)
l3 = [x for x in l1 if x not in l2set]

контрольний код тесту:

import time

l1 = list(range(1000*10 * 3))
l2 = list(range(1000*10 * 2))

l2set = {x for x in l2}

tic = time.time()
l3 = [x for x in l1 if x not in l2set]
toc = time.time()
diffset = toc-tic
print(diffset)

tic = time.time()
l3 = [x for x in l1 if x not in l2]
toc = time.time()
difflist = toc-tic
print(difflist)

print("speedup %fx"%(difflist/diffset))

результат тестування:

0.0015058517456054688
3.968189239501953
speedup 2635.179227x    

Альтернативне рішення:

reduce(lambda x,y : filter(lambda z: z!=y,x) ,[2,3,5,8],[1,2,6,8])