4. Методы split и join - Jupyter Notebook

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

Метод .split()

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

In [1]:

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

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

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

In [2]:

1 heroes

Out[2]:

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

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

In [3]:

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

2 heroes_list

Out[3]:

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

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

In [6]:

1 heroes_list[1]

Out[6]:

'Фролло'

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

.split() .

In [7]:

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

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

In [8]:

1 fheroes

Out[8]:

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

In [9]:

1 fheroes.split(",")

Out[9]:

['Фролло', 'Гренгуар']

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

In [10]:

1 nums = input("Введите два числа: ") 2 nums_list = nums.split()

Введите два числа: 1999 2022

Список получили, но он пока состоит из строк, не из чисел.

In [11]:

1 nums_list

Out[11]:

['1999', '2022']

Поправим: превратим все элементы списка в целые числа с помощью цикла и сохраним их в новый список.

In [12]:

Чуть позже мы узнаем, как более удобно (и что приятно, более быстро) сделать то же самое, но без цикла.

А пока узнаем, можно ли разбить строку по пустоте? То есть просто разделить ее на отдельные символы. Попробуем.

In [11]:

1 "1234".split("") # нельзя

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

ValueError Traceback (most recent call last) <ipython-input-11-ab6b88668550> in <module>()

----> 1 "1234".split("") # нельзя

ValueError: empty separator

Python сообщает нам, что разделитель пустой, и делить строку на части не хочет.

Раз есть метод для разбиения строки на список строк, должна быть и обратная операция ‒ для склеивания списка строк в одну большую строку. И такой метод действительно есть!

Метод .join()

Пусть у нас есть список, состоящий из имени и фамилии.

In [13]:

1l = ['Daniel', 'Lavoie']

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

In [14]:

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

Out[14]:

'Daniel Lavoie'

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

In [15]:

1 "_".join(l)

Out[15]:

'Daniel_Lavoie'

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

In [16]:

1 a = '7'

2 b = '8'

3

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

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

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

In [11]:

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

Out[11]:

'7 8'

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

In [17]:

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

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

Out[17]:

'27.09.2022'

In [19]:

1 parts = date.split(".")

2 print(parts)

3 result = "-".join(parts)

4 print(result)

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

Метод .strip()

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

In [20]:

1 # вот так

2

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

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

Out[20]:

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

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

Out[21]:

'Гренгуар'

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

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

Out[24]:

'Гренгуар '

In [25]:

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

Out[25]:

'Гренгуар'

In [26]:

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

Out[26]:

'Гренгуар'

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

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

Out[27]:

[1, 0, 9, 25]

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

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

In [28]:

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

Out[28]:

[6, 9, 6, 3]

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

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

In [34]:

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

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

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

Out[30]:

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

In [ ]:

1