Python insert и extend: вставка по индексу и добавление пачкой

append из прошлого урока кладёт элементы только в конец. Две соседние задачи он не закрывает: поставить значение на конкретное место — вторым пунктом в плане, первым в очередь — и добавить сразу готовую пачку значений, а не по одному. Для них у списка есть методы insert и extend. Оба, как и append, меняют список на месте и возвращают None, но работают по-разному, и путаница между ними — источник тихих ошибок. Разберём оба по очереди.

insert: вставка на место

Метод принимает два аргумента: индекс, куда встать, и само значение. Всё, что стояло правее, сдвигается на одну позицию вправо:

tasks = ['помыть пол', 'купить хлеб']
tasks.insert(1, 'позвонить маме')
print(tasks)
tasks.insert(0, 'выключить утюг')
print(tasks)

['помыть пол', 'позвонить маме', 'купить хлеб']
['выключить утюг', 'помыть пол', 'позвонить маме', 'купить хлеб']

Первый вызов поставил звонок маме на индекс 1 — между полом и хлебом, второй поставил утюг в начало. Вставка не заменяет, а раздвигает: длина каждый раз растёт на единицу, ни один элемент не потерялся. Индекс 0 — это всегда начало списка, откуда и растёт нумерация из урока про индексы.

insert не падает на большом индексе

Логика подсказывает: если индекс больше длины списка, Python должен ругнуться. На деле он ведёт себя тише:

tasks = ['купить хлеб']
tasks.insert(99, 'помыть пол')
print(tasks)
print(len(tasks))

['купить хлеб', 'помыть пол']
2

Индекс 99 при одном элементе не вызвал ошибки: insert молча добавил задачу в конец. Ошибку выдаёт только отрицательный сдвиг дальше начала, например insert(-99, x) — элемент ляжет в начало. Это ловушка: опечатка в индексе не падает, а тихо меняет место вставки, и список оказывается не тем, что планировалось. Проверяйте индексы по правилам урока про индексы — там же о честной ошибке IndexError при чтении.

extend: добавить пачку

extend принимает коллекцию, раскладывает её на отдельные элементы и дописывает их в конец. В корзину приехала целая доставка:

cart = ['хлеб']
cart.extend(['яйца', 'сок', 'сыр'])
print(cart)
print(len(cart))

['хлеб', 'яйца', 'сок', 'сыр']
4

Три товара легли тремя отдельными элементами, длина выросла с одного до четырёх. Разницу с append видно лучше всего на одинаковом входе — рядом, в одном коде:

a = [1, 2]
b = [1, 2]
a.append([3, 4])
b.extend([3, 4])
print(a)
print(b)
print(len(a), len(b))

[1, 2, [3, 4]]
[1, 2, 3, 4]
3 4

append кладёт в конец один предмет, extend высыпает содержимое коробки: список в аргументе разворачивается в отдельные элементы. Отсюда правило выбора: по одному значению — append, готовой коллекцией — extend. Путаница между ними даёт либо лишнюю вложенность, либо недостающие элементы, поэтому проверяйте len после пополнения — как в уроке про len.

extend раскладывает строки на буквы

extend принимает любое перебираемое, а строка перебирается по символам. Поэтому extend от строки добавляет отдельные буквы, а не одну строку:

letters = ['а', 'б']
letters.extend('вг')
print(letters)
print(len(letters))

['а', 'б', 'в', 'г']
4

Две буквы строки легли двумя элементами. Хотите положить строку как единое целое — используйте append: он не разбирает аргумент на части. Заодно extend работает и с range — пачка чисел из диапазона добавляется без цикла:

nums = [1, 2]
nums.extend(range(3, 6))
print(nums)
print(len(nums))

[1, 2, 3, 4, 5]
5

extend требует перебираемое

Число перебирать нельзя, поэтому extend от одного числа падает. Перехватим ошибку и заодно покажем правильный путь — append:

nums = [1, 2]
try:
    nums.extend(3)
except TypeError as e:
    print('Ошибка:', e)
nums.append(3)
print(nums)

Ошибка: 'int' object is not iterable
[1, 2, 3]

Сообщение говорит прямо: 'int' object is not iterable — число не перебирается, раскладывать нечего. Правило то же, что у set() из курса про множества: extend и set едят то, по чему умеет идти for. Если хочется добавить именно одно значение — это append, а не extend.

nums = [1, 2]
nums.extend(3)

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

Три способа пополнения рядом

Соберём append, insert и extend в одну таблицу — по ней удобно проверять себя, когда список растёт не так, как планировалось:

Пополнение списка теперь закрыто полностью: append для одного значения, insert для места, extend для пачки. Обратная сторона — удаление: по значению, с возвратом и по индексу. Три способа убрать элемент разбирает следующий урок.