Python进阶—map函数

map会将一个函数映射到一个输入列表的所有元素上。模板:

map(function_to_apply, list_of_inputs)

当对一个列表所有元素进行相同操作时,新手比较常用的是,把列表中所有元素一个个地传递给一个函数,并收集输出。举个栗子,求列表中所有元素的平方:

list1 = [1, 2, 3, 4, 5]
squared = []
for i in list1:
    squared.append(i**2)

map可以让我们用一种简单而漂亮得多的方式来实现。就是这样:

list2 = [1, 2, 3, 4, 5]
squared = list(map(lambda x: x**2, list2))

大多数时候,我们使用匿名函数lambdas(这个很重要,其他章节有讲到)来配合map, 所以我在上面也是这么做的。 输入列表的元素不仅可以是变量,还可以是函数!例如对某一列表元素实现自乘、自加:

def multiply(x):
    return (x*x)
def add(x):
     return (x+x)
list3 = [2, 4, 6, 8, 10]
for i in list3:
    value = map(lambda x: x(i), [multiply, add])
print(list(value))

上面print时,加了list转换,是为了python2/3的兼容性在python2中map直接返回列表,但在python3中返回迭代器,因此为了兼容python3, 需要list转换一下。输出如下:

[4, 4]
[16, 8]
[36, 12]
[64, 16]
[100, 20]

The end,Thanks!
转自 http://17coding.top

你可能感兴趣的:(Python,函数,python)