Методы .split() и .join(), списковые включения

Метод .split()

Мы уже достаточно хорошо знакомы с вводом какой-то информации с клавиатуры с помощью input(). Однако раньше мы всегда вводили с клавиатуры только один объект: одно число или одно слово... А можно ли вводить сразу несколько слов или чисел? Конечно, можно! Посмотрим.

heroes = input("Введите имена героев мюзикла Notre Dame de Paris: ")

Введите имена героев мюзикла Notre Dame de Paris: Гренгуар Фролло Эсмеральда

Выше мы ввели довольно длинную строку: перечислили имена четырех героев через пробел. Посмотрим на нее.

heroes

'Гренгуар Фролло Эсмеральда'

А теперь разобьем эту строку по пробелу, чтобы получить три отдельных имени ‒ воспользуемся методом .split().

heroes_list = heroes.split() # по умолчанию деление по пробелу heroes_list

['Гренгуар', 'Фролло', 'Эсмеральда']

Получили самый обычный список. Можем обращаться к его элементам и даже изменять их при желании:

heroes_list[1]

'Фролло'

Разбивать строку на части можно по любому символу, достаточно указать нужный символ в скобках в .split().

fheroes = input("Введите имена любимых героев NDDP через запятую: ") Введите имена любимых героев NDDP через запятую: Фролло,Гренгуар fheroes

'Фролло,Гренгуар' fheroes.split(",") ['Фролло', 'Гренгуар']

Попробуем проделать то же самое, но для чисел.

И мы хотим элементы списка склеить в одну строку так, чтобы между именем и фамилией был пробел.

" ".join(l) # вуаля

'Daniel Lavoie'

Могли бы вместо пробела в кавычках поставить любой символ, например, нижнее подчеркивание.

"_".join(l)

'Daniel_Lavoie'

Важно: метод .join() берет в качестве аргумента именно список, не просто перечисленные элементы через запятую.

a = '7' b = '8'

" ".join(a,b) # не работает

----------------------------------------------------------------------

last)

<ipython-input-16-e664b268447f> in <module> 2 b = '8'

3

----> 4 " ".join(a,b) # не работает

TypeError: join() takes exactly one argument (2 given)

" ".join([a,b]) # работает

'7 8'

Иногда удобно сочетать методы .split() и .join(). Например, перед нами стоит такая задача. Пользователь вводит дату в формате день-месяц-год через точку, а мы хотим выводить на экран дату в том же формате, но через дефис. Чтобы решить эту задачу, можно взять введенную пользователем строку с датой, разбить ее по точке, а затем склеить части, но уже используя дефис.

date = input("Введите дату: ") date

Введите дату: 27.09.2022

'27.09.2022'

parts = date.split(".") print(parts)

result = "-".join(parts) print(result)

['27', '09', '2022'] 27-09-2022

Метод .strip()

При работе со строками можно столкнуться с еще одной сложностью ‒ с пробелами. Если мы попросим пользователя ввести имена героев через запятую, перед запятой он, скорее всего, поставит пробел.

# вот так

fheroes = input("Введите имена любимых героев NDDP через запятую: ") fheroes

Введите имена любимых героев NDDP через запятую: Фролло, Гренгуар 'Фролло, Гренгуар'

Как быть? Разбить строку по запятой, а потом избавиться от пробела. Избавиться от пробела поможет метод .strip().

Метод .strip() работает по-умному: он убирает пробелы в начале и в конце строки, пробелы в середине (между словами) никуда не денутся.

'Гренгуар Фролло'

На самом деле это не случайно. Если посмотреть на все методы для строк, то можно увидеть целых три варианта со.strip(): .lstrip(), .rstrip() и

.strip(). Метод .lstrip() убирает все пробелы в начале строки (слева, l ‒ от left), метод .rstrip() ‒ все пробелы в конце строки (справа, r ‒ от right), а метод .strip()является их объединением, то есть убирает все пробелы в начале и в конце строки.

"Гренгуар ".lstrip()

'Гренгуар '

" Гренгуар ".rstrip()

'Гренгуар'

" Гренгуар ".strip()

'Гренгуар'

Списковые включения

Сейчас мы познакомимся со списковыми включениями (list comprehensions) ‒ объектами, которые позволяют создавать новые списки на основе старых, без создания пустого списка и заполнения его в цикле. Чтобы понять, что такое списковые включения, лучше сначала посмотреть на пример. Попробуем создать список, состоящий из элементов другого списка, возведенных в квадрат.

old = [1, 0, 3, 5]

new = [i**2 for i in old] new

[1, 0, 9, 25]

Цикл for у нас по-прежнему задействован, только немного в ином качестве и в другой последовательности: сначала мы указываем, что нужно делать с элементами, а потом ‒ по каким элементам будем пробегаться. В нашем примере мы говорим, что возводим элементы i в квадрат, а потом сообщаем, что эти элементы i мы берем из списка old.

Вернемся к примеру с числами, введенными с клавиатуры, и попробуем получить список из чисел, используя списковые включения.

snumbers = input("Введите числа через пробел: ") lnumbers = snumbers.split()

numbers = [int(n) for n in lnumbers] # делаем все элементы целочисленными

numbers

Введите числа через пробел: 6 9 6 3

[6, 9, 6, 3]

Использование списковых включений ‒ не только более компактный, но и более быстрый способ создания новых списков на основе старых. Давайте сравним время выполнения кода с циклом for и кода со списковыми включениями.

%%timeit

#timeit - магическое слово Jupyter, magic command,

#которое позволяет измерить время исполнения ячейки с кодом

#ячейка запускается много раз, и выводится среднее время выполнения со стандарным отклонением

#должно быть указано первой строчкой в ячейке

L = []

for i in range(1, 10001): L.append(i**2)

5.4 ms ± 7.57 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

%%timeit

L = [j**2 for j in range(1, 10001)]

4.55 ms ± 18.3 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

А теперь напишем компактный и быстрый код, который будет разбивать строку по запятой и при этом возвращать имена героев без лишних пробелов.

names = input("Введите имена героев через запятую: ") lnames = names.split(",")

print(lnames)

names_final = [name.strip() for name in lnames] # берем все элементы и отсекаем пробелы

names_final # готово!

Введите имена героев через запятую: bob, nika, mom ['bob', ' nika', ' mom']

['bob', 'nika', 'mom']