- PVSM.RU - https://www.pvsm.ru -
Привет. Часто при работе с последовательностями встает вопрос об их создании. Вроде бы привык использовать списковые включения (List Comprehension), а в книжках кричат об обязательном использовании встроенной функции map.
В этой статье мы рассмотрим эти подходы к работе с последовательностями, сравним производительность, а также определим в каких ситуациях какой подход лучше.
Списковое включение — это встроенный в Python механизм генерации списков. У него только одна задача — это построить список. Списковое включение строит список из любого итерируемого типа, преобразуя (фильтруя) поступаемые значения.
Пример простого спискового включения для генерации списка квадратов чисел от 0 до 9:
squares = [x*x for x in range(10)]
Результат:
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
map — встроенная в язык функция. Принимает на вход в качестве первого параметра — функцию, а в качестве второго — итерируемый объект. Возвращает генератор (Python 3.x) или список (Python 2.x). Я остановлю свой выбор на Python 3.
Пример вызова функции map для генерации списка квадратов чисел от 0 до 9:
squares = list(map(lambda x: x*x, range(10)))
Результат:
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
В качестве эксперимента будем считать квадраты чисел из промежутка от 0 до 9,999,999:
python -m timeit -r 10 "[x*x for x in range(10000000)]"
python -m timeit -r 10 "list(map(lambda x: x*x, range(10000000)))"
Результаты:
1 loop, best of 10: 833 msec per loop
1 loop, best of 10: 1.22 sec per loop
Как видим способ с List Comprehension работает примерно на 32% быстрее. Продизассемблировав не удается получить полных ответов, так как функция map «как будто скрывает детали своей работы». Но скорее всего это связано с постоянным вызовом lambda функции, внутри которой уже делаются вычисления квадрата. В случае с List Comprehension нам требуются только вычисления квадрата.
Для сравнения будем также считать квадраты чисел, но вычисления теперь будут находиться внутри функции:
python -m timeit -r 10 -s "def pow2(x): return x*x" "[pow2(x) for x in range(10000000)]"
python -m timeit -r 10 -s "def pow2(x): return x*x" "list(map(pow2, range(10000000)))"
Результаты:
1 loop, best of 10: 1.41 sec per loop
1 loop, best of 10: 1.21 sec per loop
В этот раз ситуация обратная. Метод с использованием map оказался на 14% быстрее. В данном примере оба метода оказываются в одинаковой ситуации. Оба должны вызывать функцию для вычисления квадрата. Однако внутренние оптимизации функции map позволяют ей показывать лучшие результаты.
Ниже представлено правило выбора нужного способа:
Возможно, существуют исключения из этого правило, но в большинстве случаев оно поможет вам сделать правильный выбор!
Почему же многие призывают использовать map. Дело в том, что в некоторых случаях map действительно безопаснее List Comprehension.
Например:
symbols = ['a', 'b', 'c']
values = [1, 2, 3]
for x in symbols:
print(x)
squared = [x*x for x in values] # Все плохо. "x" из верхнего "for" затёрт
print(x)
Вывод программы будет следующим:
a
3
b
3
c
3
Теперь перепишем этот же код с использованием map:
symbols = ['a', 'b', 'c']
values = [1, 2, 3]
for x in symbols:
print(x)
squared = map(lambda x: x*x, values) # Все норм, это локальная область видимости
print(x)
Вывод:
a
a
b
b
c
c
Самые наблюдательные уже могли заметить по синтаксису использования map, что это Python 2. Действительно, во втором питоне была подобного рода проблема с затиранием переменных. Однако в Python 3 эта проблема была исправлена и больше не является актуальной.
Описанные выше примеры покажут одинаковые результаты. Может также показаться, что это глупая ошибка и ты такую никогда не допустишь, однако это может произойти когда вы просто перенесли блок кода с внутренним циклом из другого блока. Такая ошибка может потратить у Вас кучу времени и нервов на её исправление.
Сравнение показало, что каждый из способов хорош в своей ситуации.
P.S.: Если я что-то упустил, то с радостью готов обсудить это вместе с вами в комментариях.
Автор: megagnom37
Источник [1]
Сайт-источник PVSM.RU: https://www.pvsm.ru
Путь до страницы источника: https://www.pvsm.ru/python/339286
Ссылки в тексте:
[1] Источник: https://habr.com/ru/post/479252/?utm_source=habrahabr&utm_medium=rss&utm_campaign=479252
Нажмите здесь для печати.