Зарегистрируйтесь для доступа к 15+ бесплатным курсам по программированию с тренажером

Отображение (map) Python: Функции

Рассмотрим следующую задачу. Возьмём словарь пользователей и извлечём из него имена всех пользователей:

users = [
  { 'name': 'Igor', 'age': 19 },
  { 'name': 'Danil', 'age': 1 },
  { 'name': 'Vovan', 'age': 4 },
  { 'name': 'Matvey', 'age': 16 },
]

result = []
for user in users:
  result.append(user['name'])

print(result) # => ['Igor', 'Danil', 'Vovan', 'Matvey']

Здесь мы видим обычную агрегацию с использованием цикла for...in. А что, если нам понадобится извлечь возраст? Повторяем:

result = []
for user in users:
  result.append(user['age'])

print(result) # => [19, 1, 4, 16]

В примерах выше легко увидеть закономерность. Выполняется один и тот же проход по циклу, и результат собирается в список result. Единственное, что меняется — значение, которое мы извлекаем из элементов исходного словаря.

Операция, которую мы выполняли в обеих ситуациях, называется отображением (mapping). В коде мы взяли исходную коллекцию и отобразили ее в другую коллекцию, попутно выполнив необходимые преобразования над каждым элементом. Важно, что размер получившейся коллекции равен размеру исходной.

Задача отображать данные в реальном коде встречается буквально на каждом шагу. Это настолько важная операция, что для нее создана специальная функция высшего порядка map():

names = map(lambda user: user['name'], users)
list(names) # => <map object at 0x71de31ee0250>

for name in names:
  print(name)

# => Igor
# => Danil
# => Vovan
# => Matvey

Функция map() принимает первым параметром функцию, а последующими коллекцию. Дальше, внутри себя, map() перебирает элементы переданной коллекции и для каждого элемента вызывает переданную функцию. Возвращает map() уже знакомый вам итератор.

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

Вместо перебора в цикле, существует более простой способ извлечь все элементы из итератора - применить конструктор list()

names = map(lambda user: user['name'], users)
list(names) # ['Igor', 'Danil', 'Vovan', 'Matvey']

ages = map(lambda user: user['age'], users)
list(ages) # [19, 1, 4, 16]

# или зададим колбек отдельно
callback = lambda user: user['age']
list(map(callback, users)) # [19, 1, 4, 16]

Сравните решение задачи получения списка имен через цикл и с помощью map(). Последний имеет довольно много преимуществ. Во-первых, код с ним значительно короче. Во-вторых, от нас скрыта повторяющаяся логика перебора. Больше не нужно явно определять цикл и выполнять руками все те операции, которые можно не выполнять. Функция map() позволяет сосредоточиться на сути происходящего, скрывая ненужные детали (проход по циклу).

Типичный пример, который любят приводить в документации к функции map() разных языков программирования — применение некоторой арифметической операции к каждому элементу коллекции:

numbers = [5, 2, 3]

new_numbers = map(lambda number: number ** 2, numbers)
list(new_numbers) # => [25, 4, 9]

new_numbers_2 = map(lambda number: number + 3, numbers)
list(new_numbers_2) # => [8, 5, 6]

Пример выглядит искусственно, но хорошо отражает суть операции.

Реализация

Напишем свою собственную функцию my_map(), работающую аналогично встроенной функции map():

def my_map(callback, collection):
  for item in collection:
    # Вызов переданного колбека на каждом элементе коллекции
    new_item = callback(item)
    # Сделаем нашу функцию тоже ленивой с помощью yield
    yield new_item

numbers = [5, 2, 3]
new_numbers = my_map(lambda number: number ** 2, numbers)
list(new_numbers) # [25, 4, 9]

Главное отличие функции my_map() (и встроенной map()) от ручного обхода коллекции заключается в том, что функция my_map() не знает, что нужно сделать с каждым элементом коллекции. Поэтому она принимает функцию-колбек, которую вызывает для каждого элемента исходного коллекции. Чем будет этот результат — функция my_map() не знает, и ей этого знать не нужно. Ответственность за обработку лежит на пользователях.


Дополнительные материалы

  1. Функция map

Аватары экспертов Хекслета

Остались вопросы? Задайте их в разделе «Обсуждение»

Вам ответят команда поддержки Хекслета или другие студенты

Об обучении на Хекслете

Для полного доступа к курсу нужен базовый план

Базовый план откроет полный доступ ко всем курсам, упражнениям и урокам Хекслета, проектам и пожизненный доступ к теории пройденных уроков. Подписку можно отменить в любой момент.

Получить доступ
1000
упражнений
2000+
часов теории
3200
тестов

Открыть доступ

Курсы программирования для новичков и опытных разработчиков. Начните обучение бесплатно

  • 130 курсов, 2000+ часов теории
  • 1000 практических заданий в браузере
  • 360 000 студентов
Отправляя форму, вы принимаете «Соглашение об обработке персональных данных» и условия «Оферты», а также соглашаетесь с «Условиями использования»

Наши выпускники работают в компаниях:

Логотип компании Альфа Банк
Логотип компании Aviasales
Логотип компании Yandex
Логотип компании Tinkoff
Рекомендуемые программы
профессия
Обучитесь разработке бэкенда сайтов и веб-приложений — серверной части, которая отвечает за логику и базы данных
10 месяцев
с нуля
Старт 7 ноября

Используйте Хекслет по-максимуму!

  • Задавайте вопросы по уроку
  • Проверяйте знания в квизах
  • Проходите практику прямо в браузере
  • Отслеживайте свой прогресс

Зарегистрируйтесь или войдите в свой аккаунт

Отправляя форму, вы принимаете «Соглашение об обработке персональных данных» и условия «Оферты», а также соглашаетесь с «Условиями использования»