Здесь может быть очень полезна функция map
. Используя map
, мы можем применить любую функцию к каждому элементу итерации.
Python 3.x
>>> def my_mul(x,y):
... return x*y
...
>>> a = [1,2,3,4]
>>> b = [2,3,4,5]
>>>
>>> list(map(my_mul,a,b))
[2, 6, 12, 20]
>>>
Конечно:
map(f, iterable)
эквивалентно
[f(x) for x in iterable]
Итак, мы можем получить наше решение через:
>>> [my_mul(x,y) for x, y in zip(a,b)]
[2, 6, 12, 20]
>>>
В Python 2.x map()
означает: применить функцию к каждому элементу итерации и построить новый список. В Python 3.x map
создают итераторы вместо списков.
Вместо my_mul
можно использовать оператор mul
Python 2.7
>>>from operator import mul # import mul operator
>>>a = [1,2,3,4]
>>>b = [2,3,4,5]
>>>map(mul,a,b)
[2, 6, 12, 20]
>>>
Python 3.5+
>>> from operator import mul
>>> a = [1,2,3,4]
>>> b = [2,3,4,5]
>>> [*map(mul,a,b)]
[2, 6, 12, 20]
>>>
Обратите внимание, что, поскольку map()
создает итератор, мы используем *
итеративный оператор распаковки для получения списка. Подход к распаковке немного быстрее, чем конструктор list
:
>>> list(map(mul,a,b))
[2, 6, 12, 20]
>>>
person
sg7
schedule
13.01.2018