Перевод статьи «4 Ways To Level Up Your Python Code».
Прочитав эту статью, вы научитесь избегать ненужных циклов for, более эффективно получать доступ к элементам словаря и многому другому!
Один из плюсов Python — с его помощью можно заставить работать что угодно.
Еще один плюс — множество скрытых фишек, которые не только облегчают написание кода, но и делают код «чище» и удобнее для чтения.
В этой статье мы разберем фишки, которые помогут вам отточить ваши навыки.
1. Забудьте о range(len()). Наш выбор — enumerate()
Часто возникают ситуации, когда нужно перебрать коллекцию и отслеживать индексы элементов, к которым вы получили доступ. Для этих целей многие использует range(len(коллекция)).
Функция enumerate возвращает enumerate-объект, который хранит в себе и индексы элементов, и их значения.
С помощью этой функции вы облегчите доступ и к индексам, и к значениям.
Рассмотрим пример, в котором видна разница между range(len(коллекция)) и enumerate(коллекция).
# объявляем коллекцию — список:
names = ['Никита', 'Дарья', 'Екатерина', 'Дмитрий', 'Владимир']
# в случае с range(len(collection)) вы бы написали:
for i in range(len(names)):
print(i, names[i])
# в случае с enumerate вы бы написали:
for idx, name in enumerate(names):
print(idx, name)
# Оба варианта возвращают:
# 0 Никита
# 1 Дарья
# 2 Екатерина
# 3 Дмитрий
# 4 Владимир
Еще одна полезная особенность enumerate — можно начинать отсчет не с 0. Для этого вам нужно лишь добавить параметр — start=параметр. Предположим, мы хотим начать выводить элементы списка с 1.
# объявляем коллекцию — список:
names = ['Никита', 'Дарья', 'Екатерина', 'Дмитрий', 'Владимир']
# при использовании enumerate перебор с 1 реализуется так:
for idx, name in enumerate(names, start=1):
print(idx, name)
# Цикл вернет:
# 1 Никита
# 2 Дарья
# 3 Екатерина
# 4 Дмитрий
# 5 Владимир
2. Забудьте о квадратных скобках — получайте элементы словаря с помощью .get()
Допустим, мы объявили словарь:
nikita = {
'возраст':32,
'пол':'мужской',
'трудоустроен':True,
}
И хотим получить значение возраст. Возможно, вы уже привыкли использовать в таких случаях nikita['возраст']. В нашем примере это сработает, но если значения не существует, — нет.
Например, при попытке получить доступ к несуществующему значению 'место жительства' мы получим ошибку KeyError:
nikita = {
'возраст':32,
'пол':'мужской',
'трудоустроен':True,
}
print(nikita['место жительства'])
# Вывод:
# KeyError: 'место жительства'
Если бы мы перебирали список словарей, эта ошибка могла бы прервать выполнение нашей программы. Да, конечно, мы бы могли использовать try, except — но зачем, если есть метод .get()?
Если значения, к которому мы хотим получить доступ, не существует — возвращается None. Это очень удобно!
nikita = {
'возраст':32,
'пол':'мужской',
'трудоустроен':True,
}
print(nikita.get('место жительства'))
# Вывод:
# None
3. Облегчите перебор нескольких списков с помощью zip()
Порой возникают ситуации, когда у нас есть несколько коллекций (например, списков) и все их нужно перебрать.
В примере №1 мы разбирали способ с применением range(len(collection)), но он не самый лучший.
Рассмотрим пример:
names = ['Никита', 'Дарья', 'Екатерина', 'Дмитрий']
ages = [32, 28, 37, 53]
gender = ['Мужчина', 'Женщина', 'Женщина', 'Мужчина']
# старый и скучный способ:
for_looped = []
for i in range(len(names)):
for_looped.append((names[i], ages[i], gender[i]))
print(for_looped)
# Вывод:
# [('Никита', 32, 'Мужчина'), ('Дарья', 28, 'Женщина'), ('Екатерина', 37, 'Женщина'), ('Дмитрий', 53, 'Мужчина')]
Как видите, в результате мы получаем список, содержащий кортежи с элементами из каждого списка.
Этот способ, конечно, рабочий. Но если коллекций станет слишком много, код выйдет громоздким.
С помощью встроенной функции zip() мы можем все немного упростить.
Обратите внимание: функция zip() возвращает zip-объект, но с помощью приведения типов вы можете преобразовать его — например, с помощью list(), tuple() или dict().
А теперь рассмотрим тот же пример и снова вернем список кортежей:
names = ['Никита', 'Дарья', 'Екатерина', 'Дмитрий']
ages = [32, 28, 37, 53]
gender = ['Мужчина', 'Женщина', 'Женщина', 'Мужчина']
# перебор списков с помощью zip()
zipped = zip(names, ages, gender)
zipped_list = list(zipped)
print(zipped_list)
# Вывод:
# [('Никита', 32, 'Мужчина'), ('Дарья', 28, 'Женщина'), ('Екатерина', 37, 'Женщина'), ('Дмитрий', 53, 'Мужчина')]
Выглядит намного лучше, не так ли?
Также вы можете преобразовывать коллекции в словари напрямую. Но обратите внимание — коллекций должно быть только две.
Например, вы хотите создать словарь, в котором пара ключ-значение — это имя:возраст. Сделать это можно так:
names = ['Никита', 'Дарья', 'Екатерина', 'Дмитрий']
ages = [32, 28, 37, 53]
gender = ['Мужчина', 'Женщина', 'Женщина', 'Мужчина']
ages = dict(zip(names,ages))
print(ages)
# Вывод:
# {'Никита': 32, 'Дарья': 28, 'Екатерина': 37, 'Дмитрий': 53}
Если вы сделаете то же самое, но коллекций будет больше двух — появится исключение ValueError.
4. Используем f-строки для упрощения вывода информации в консоль
Начиная с Python 3.8 f-строки могут быть использованы для самодокументирования — с помощью =.
Это значит, что для того, чтобы вывести в консоль какое-то значение, вам больше не нужно писать f"variable_name = {variable_name}".
Рассмотрим пример:
some_variable = "ПРИВЕТ!"
print(f"some_variable={some_variable}")
# Вывод:
# some_variable=ПРИВЕТ!
Вместо этого вы можете написать:
some_variable = 'Привет!'
print(f'{some_variable=}')
# Вывод:
# some_variable='Привет!'
Изменение это небольшое, но довольно важное. Благодаря ему вы не просто избавитесь от повторений в коде, но и упростите его чтение.
Заключение
Возьмите все эти четыре совета на заметку : благодаря им ваш код будет стабильнее. Еще один плюс — ваш код будет выглядеть гораздо совершеннее и утонченнее.
