add и update в Python: добавление элементов в set

Созданное множество — не приговор: оно изменяемое, и элементы в него добавляют всю жизнь программы. У этой операции два глагола: add для одиночного значения и update для целой пачки. Разница кажется мелкой, но у update спрятаны две ловушки — строки разбираются на буквы, а возвращаемое значение у обоих методов отсутствует. Разберём всё на живом коде.

add: один элемент

Метод add принимает один аргумент и кладёт его в множество. Наполним теги статьи по одному:

tags = {'python'}
tags.add('code')
tags.add('datetime')
print(sorted(tags))
print(len(tags))

['code', 'datetime', 'python']
3

Каждый вызов add вырос множество на один элемент. Заметьте: add меняет множество на месте — его не нужно присваивать, теги после вызова уже новые. Напечатать по-прежнему честнее через sorted: порядок внутри set не гарантирован.

Дубликат тихо игнорируется

Что произойдёт, если добавить элемент, который уже есть? Ничего — и это не сбой, а дизайнерское решение: уникальность встроена в структуру:

tags = {'python', 'code'}
before = len(tags)
tags.add('python')
after = len(tags)
print('было:', before, '- стало:', after)
print(sorted(tags))

было: 2 - стало: 2
['code', 'python']

Никакой ошибки, никакого изменения длины — повтор просто не пустили внутрь. Та же тишина и у update: он вольёт новый набор, и дубликаты из него растворятся на границе множества:

base = {1, 2}
base.update([2, 3, 2])
print(sorted(base))
print(len(base))

[1, 2, 3]
3

Из трёх значений списка в множество попали два новых: повторная двойка исчезла. Писать проверку есть ли элемент перед add не нужно — множество само решает, пускать ли повтор.

update: пачка элементов

add умеет один элемент. Когда приходит целая поставка — список тегов, результаты запроса, другой журнал — вызывать add в цикле не нужно: метод update принимает набор целиком:

tags = {'python'}
tags.update(['code', 'datetime'])
print(sorted(tags))
print(len(tags))

['code', 'datetime', 'python']
3

Два элемента списка легли в множество одним вызовом. У update есть ещё одна суперсила: он принимает сразу несколько аргументов и вольёт их все — списки, множества, что угодно перебираемое:

base = {1, 2}
base.update([3, 4], {5})
print(sorted(base))

[1, 2, 3, 4, 5]

Один вызов, два источника, пять элементов: список и литеральное множество влились за раз. Именно поэтому update — не пара к add, а отдельный инструмент: он про источники, а не про элементы.

Ловушка: update от строки

Строка — перебираемая величина: по ней можно идти циклом. Значит, update её примет — но добавит не строку, а её буквы по одной. Это классическая ловушка новичка:

letters = {'x'}
letters.update('abc')
print(sorted(letters))

['a', 'b', 'c', 'x']

В множество легли три буквы: a, b, c — по символу из строки. Если нужна строка целиком, как один элемент, используют add: add('abc') положил бы единственное значение 'abc'. Правило различения простое: add кладёт аргумент как есть, update разбирает аргумент на элементы.

Списки, множества и строки update принимает; а вот число перебрать нельзя — update, как и set(), требует перебираемое. Попытка скормить ему число падает прямо на месте:

tags = {'python'}
tags.update(42)

Traceback (most recent call last):
  File "app.py", line 2, in <module>
    tags.update(42)
    ~~~~~~~~~~~^^^^
TypeError: 'int' object is not iterable

Сообщение то же, что у set(42) из прошлого урока: объект не перебирается. Число добавляют через add — оно элемент, а не набор.

add и update ничего не возвращают

Оба метода меняют множество на месте и возвращают None — никакое. Это чужой для списков текст: append у списка ведёт себя точно так же, но рука всё равно тянется написать присваивание. Посмотрим, что вернёт add:

s = {1}
result = s.add(2)
print(result)
print(sorted(s))

None
[1, 2]

Элемент добавился, а result — None: метод отработал внутри множества и вернул пустоту. Отсюда главная ошибка темы: s = s.add(2) записывает в s значение None, и множество теряется — все элементы исчезают вместе с именем. Методы, меняющие объект на месте, не возвращают этот объект: они уже всё сделали.

Шпаргалка: add и update

Два метода, два масштаба. Таблица помогает выбрать за секунду:

Пополнять множество научились — один элемент и пачками. Обратная операция не менее важна: убрать элемент можно тремя способами, и один из них падает на отсутствующем значении. Об этом — урок про remove, discard и pop.