Курс Python → Применение функции к списку

Для применения функции ко всем элементам списка в Python можно воспользоваться функцией map(). Эта функция принимает два аргумента: функцию, которую нужно применить, и список, к каждому элементу которого нужно применить эту функцию. Например, если у нас есть список чисел [1, 2, 3, 4, 5] и мы хотим возвести каждое число в квадрат, мы можем использовать функцию map().

Пример кода:


numbers = [1, 2, 3, 4, 5]
squared_numbers = list(map(lambda x: x**2, numbers))
print(squared_numbers)

В этом примере мы создаем список numbers и затем с помощью функции map() применяем к каждому элементу этого списка лямбда-функцию, которая возводит число в квадрат. Результатом выполнения этого кода будет список [1, 4, 9, 16, 25].

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

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

Твои коллеги будут рады, поделись в