Как добавить содержимое итерируемого в набор?


Ответы:


228

Вы можете добавить элементы listк setтак:

>>> foo = set(range(0, 4))
>>> foo
set([0, 1, 2, 3])
>>> foo.update(range(2, 6))
>>> foo
set([0, 1, 2, 3, 4, 5])

2
Просто оглянулся на мой сеанс интерпретатора, и я действительно попробовал это, но подумал, что он добавил весь список как элемент набора из-за квадратных скобок в представлении набора. Я никогда раньше не замечал, что они так представлены.
Ян Маккиннон

7
Это представление позволяет вам вставить его обратно в интерактивную сессию, потому что setконструктор принимает в качестве аргумента итерацию.
Фрэнк Кастерс

3
Обратите внимание, что представление просто, например, {1, 2, 3}в Python 3, тогда как это было set([1, 2, 3])в Python 2.
Радон Росборо

40

В интересах любого, кто может поверить, например, что выполнение aset.add()в цикле будет иметь производительность, конкурентоспособную по сравнению с выполнением aset.update(), вот пример того, как вы можете быстро проверить свои убеждения перед публикацией:

>\python27\python -mtimeit -s"it=xrange(10000);a=set(xrange(100))" "a.update(it)"
1000 loops, best of 3: 294 usec per loop

>\python27\python -mtimeit -s"it=xrange(10000);a=set(xrange(100))" "for i in it:a.add(i)"
1000 loops, best of 3: 950 usec per loop

>\python27\python -mtimeit -s"it=xrange(10000);a=set(xrange(100))" "a |= set(it)"
1000 loops, best of 3: 458 usec per loop

>\python27\python -mtimeit -s"it=xrange(20000);a=set(xrange(100))" "a.update(it)"
1000 loops, best of 3: 598 usec per loop

>\python27\python -mtimeit -s"it=xrange(20000);a=set(xrange(100))" "for i in it:a.add(i)"
1000 loops, best of 3: 1.89 msec per loop

>\python27\python -mtimeit -s"it=xrange(20000);a=set(xrange(100))" "a |= set(it)"
1000 loops, best of 3: 891 usec per loop

Похоже, что цена за элемент в циклическом подходе в три раза выше, чем в updateподходе.

Использование |= set()стоит примерно в 1,5 updateраза больше, чем половина добавления каждого отдельного элемента в цикле.


14

Вы можете использовать функцию set (), чтобы преобразовать итерируемое в набор, а затем использовать стандартный оператор обновления набора (| =), чтобы добавить уникальные значения из вашего нового набора в существующий.

>>> a = { 1, 2, 3 }
>>> b = ( 3, 4, 5 )
>>> a |= set(b)
>>> a
set([1, 2, 3, 4, 5])

5
.updateПреимущество использования заключается в том, что аргумент может быть любым итеративным - не обязательно набором - в отличие от RHS |=оператора в вашем примере.
tzot

1
Хорошая точка зрения. Это просто эстетический выбор, так как set () может преобразовать итерируемое в набор, но количество нажатий клавиш одинаково.
Gbc

Я никогда не видел этого оператора раньше, я буду рад использовать его, когда он появится в будущем; Спасибо!
eipxen

1
@eipxen: есть |для объединения, &для пересечения и ^для получения элементов, которые находятся в одном или другом, но не в обоих. Но в динамически типизированном языке, где иногда трудно читать код и знать типы объектов, летающих вокруг, я не решаюсь использовать эти операторы. Кто-то, кто их не распознает (или, возможно, даже не понимает, что Python допускает такие операторы), может быть сбит с толку и думать, что происходят какие-то странные побитовые или логические операции. Было бы хорошо, если бы эти операторы работали и на других итерируемых элементах ...
ArtOfWarfare

Провел несколько тестов времени на это .update()и добавил отдельные элементы в цикл. Нашел, что .update()было быстрее. Я добавил свои результаты в этот существующий ответ: stackoverflow.com/a/4046249/901641
ArtOfWarfare

4

Просто быстрое обновление, время использования Python 3:

#!/usr/local/bin python3
from timeit import Timer

a = set(range(1, 100000))
b = list(range(50000, 150000))

def one_by_one(s, l):
    for i in l:
        s.add(i)    

def cast_to_list_and_back(s, l):
    s = set(list(s) + l)

def update_set(s,l):
    s.update(l)

результаты:

one_by_one 10.184448844986036
cast_to_list_and_back 7.969255169969983
update_set 2.212590195937082

0

Используйте понимание списка.

Короткое замыкание создания итерируемого с использованием списка, например :)

>>> x = [1, 2, 3, 4]
>>> 
>>> k = x.__iter__()
>>> k
<listiterator object at 0x100517490>
>>> l = [y for y in k]
>>> l
[1, 2, 3, 4]
>>> 
>>> z = Set([1,2])
>>> z.update(l)
>>> z
set([1, 2, 3, 4])
>>> 

[Изменить: пропустил заданную часть вопроса]


1
Я не вижу никаких наборов? Я что-то упускаю?
Ян Маккиннон

-2
for item in items:
   extant_set.add(item)

Для протокола, я думаю, что утверждение «Должен быть один - и желательно только один - очевидный способ сделать это». подделка Предполагается, что многие технические люди думают, что все думают одинаково. То, что очевидно для одного человека, не так очевидно для другого.

Я бы сказал, что предложенное мной решение легко читается и выполняет то, что вы просите. Я не верю, что с этим связаны какие-то проблемы с производительностью, хотя я допускаю, что я что-то упустил. Но, несмотря на все это, это может быть неочевидным и предпочтительным для другого разработчика.


Argh! Цикл for в одной строке, как это, форматирует в моем ответе - я бы никогда этого не сделал. Когда-либо.
Джейдел

Вы абсолютно правы. Я отредактировал пост, чтобы исправить мои повреждения. Спасибо :)
Джейдел

9
Вам не хватает точки, которая aset.update(iterable)зацикливается на скорости C, тогда как for item in iterable: aset.add(item)зацикливается на скорости Python, с поиском метода и вызовом метода (aarrgghh !!) для каждого элемента.
Джон Мачин

1
Извините, он ничего не сказал о производительности в своем вопросе, поэтому я не беспокоился об этом.
Джейдель
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.