Лучший способ найти пересечение нескольких множеств?

5 ответов

• 332 рейтинг

  Из Python версии 2. 6 на вы можете использовать несколько аргументов для set.intersection() , как

  u = set.intersection(s1, s2, s3)
  

  Если наборы находятся в списке, это означает:

  u = set.intersection(*setlist)
  

  , где *a_list - расширение списка

  ответ дан sth, с репутацией 160276, 29.03.2010
• 54 рейтинг

  По состоянию на 2. 6, set.intersection занимает сколько угодно итераций.

  >>> s1 = set([1, 2, 3])
  >>> s2 = set([2, 3, 4])
  >>> s3 = set([2, 4, 6])
  >>> s1 & s2 & s3
  set([2])
  >>> s1.intersection(s2, s3)
  set([2])
  >>> sets = [s1, s2, s3]
  >>> set.intersection(*sets)
  set([2])
  

  ответ дан Mike Graham, с репутацией 48069, 29.03.2010
• 14 рейтинг

  Ясно, что set.intersection - это то, что вы хотите здесь, но в случае, если вам когда-нибудь понадобится обобщение «взять сумму всех этих», «взять произведение всех этих», «взять xor всех этих», то, что вы ищете является функцией reduce:

  from operator import and_
  from functools import reduce
  print(reduce(and_, [{1,2,3},{2,3,4},{3,4,5}])) # = {3}
  

  или

  print(reduce((lambda x,y: x&y), [{1,2,3},{2,3,4},{3,4,5}])) # = {3}
  

  ответ дан Thomas Ahle, с репутацией 20791, 29.02.2012
• 11 рейтинг

  Если у вас нет Python 2. 6 или выше, альтернативой является написание явного цикла for:

  def set_list_intersection(set_list):
    if not set_list:
      return set()
    result = set_list[0]
    for s in set_list[1:]:
      result &= s
    return result
  
  set_list = [set([1, 2]), set([1, 3]), set([1, 4])]
  print set_list_intersection(set_list)
  # Output: set([1])
  

  Вы также можете использовать reduce:

  set_list = [set([1, 2]), set([1, 3]), set([1, 4])]
  print reduce(lambda s1, s2: s1 & s2, set_list)
  # Output: set([1])
  

  Однако многим программистам Python это не нравится, , включая самого Гвидо :

  Около 12 лет назад Python приобрел лямбду-редуктор (Reduce (), Filter () и Map (), любезно (я полагаю) хакер Lisp, который пропустил их и представил рабочие исправления). Но, несмотря на значение PR, я думаю, что эти функции должны быть вырезаны из Python 3000.

  Так что теперь уменьшаем (). Это на самом деле тот, который я всегда ненавидел больше всего, потому что, за исключением нескольких примеров, включающих + или *, почти каждый раз, когда я вижу вызову redu () с нетривиальным аргументом функции, мне нужно взять ручку и бумагу, чтобы представьте диаграмму, что на самом деле подается в эту функцию, прежде чем я пойму, что должен делать redu (). Поэтому, на мой взгляд, применимость метода limit () в значительной степени ограничена ассоциативными операторами, и во всех других случаях лучше явно выписать цикл накопления.

  ответ дан Ayman Hourieh, с репутацией 84055, 29.03.2010
• 1 рейтинг

  Здесь я предлагаю универсальную функцию для пересечения нескольких множеств, пытаясь воспользоваться лучшим доступным методом:

  def multiple_set_intersection(*sets):
      """Return multiple set intersection."""
      try:
          return set.intersection(*sets)
      except TypeError: # this is Python < 2.6 or no arguments
          pass
  
      try: a_set= sets[0]
      except IndexError: # no arguments
          return set() # return empty set
  
      return reduce(a_set.intersection, sets[1:])
  

  Гвидо может не нравиться reduce, но мне это нравится :)

  ответ дан tzot, с репутацией 61461, 31.03.2010