Функция map для итерации объектов в Python
Пусть у нас есть функция square
для возведения чисел в квадрат. И есть
список, к элементам, которого
нужно применить данную
функцию:
def square(num):
return num ** 2
lst = [2, 3, 6, 8, 15]
В Python чтобы решить данную задачу,
можно использовать специальную функцию
map
. Она принимает параметрами функцию
и список, к элементам которого
ее нужно применить. Давайте воспользуемся
map
для решения примера:
res = map(square, lst)
print(res)
Каждый список как и любой сложный объект занимает много места в системной памяти Python. Поэтому для экономии ресурсов после выполнения кода вернется не новый список, а специальный итерируемый map-объект:
<map object at 0x000001F16674BA00>
Давайте переберем его циклом:
for el in res:
print(el)
В результате выведутся все элементы нового списка:
4
9
36
64
225
Чтобы создать из map-объекта
новый список, нужно применить
к нему функцию
list
:
lst = [2, 3, 6, 8, 15]
res = map(square, lst)
В результате выведется следующий список:
[4, 9, 36, 64, 225]
Также при работе с функцией map
в первом параметре можно задать
лямбда-функцию. Давайте перепишем предыдущий пример
с помощью лямбда-функции:
res = map(lambda num: num ** 2, lst, lst)
print(list(res))
Перепишите следующий код через лямбда-функцию:
def func(num):
return num + 1
lst = [1, 2, 3, 4, 5]
res = map(func, lst)
print(list(res))
Перепишите следующий код через лямбда-функцию:
def func(txt):
return txt[::-1]
lst = ['123', '456', '789']
res = map(func, lst)
print(list(res))