[python] Python map函数总结

Python map函数总结

本文主要介绍如何使用Python(Python3版本)的内置map()函数。简单来说map()函数会将指定的函数依次作用于某个序列的每个元素,并返回一个迭代器对象。map语法如下,其中function表示我们指定的函数,iterable表示要作用的序列,这个序列可以是一个也可以是多个。

map(function, iterable, ...)

下面实例具体介绍map()的使用方法。关于map函数的更多使用介绍见本文参考部分。

1 使用

1.1 基础示例

应用map()在整数列表上

# 定义一个函数
def square(x): 

    return x**2 

# 序列
nums = [1, 2, 3, 4, 5] 

# 对nums序列每个数求平方,返回迭代器
nums_squared = map(square, nums) 

# 输出结果
for num in nums_squared: 

    print(num)
1
4
9
16
25

如果要实现和以上map函数同等功能,需要自定义函数,例子如下,

# 定义一个函数
def square(x):

    return x * x

# 序列
def mymap(func, iterable):
    for i in iterable:
        yield func(i)

nums = [1, 2, 3, 4, 5]

# 对nums序列每个数求平方,返回迭代器
nums_squared = mymap(square, nums)

for num in nums_squared:

    print(num)
1
4
9
16
25

当然我们也可以将上面例子中的square函数用lambda代码,例子如下:

nums = [1, 2, 3, 4, 5] 

nums_squared = map(lambda x: x*x, nums) 

for num in nums_squared: 
    print(num)
1
4
9
16
25

1.2 具有多个可迭代对象的 Python 映射

我们可以将多个可迭代对象传递给map()函数,然后map函数将这两个迭代对象传入指定的函数。注意map函数必须采用与可迭代对象一样多的参数,具体如下所示:

# 定义一个函数
# 该函数必须采用两个参数,因为有两个可迭代对象传递给map()
def multiply(x, y):

    return x * y

# 序列1
nums1 = [1, 2, 3, 4, 5]
# 序列2
nums2 = [6, 7, 8, 9, 10]

# 每次从nums1和nums2中取值
mult = map(multiply, nums1, nums2)

for num in mult:
    print(num)
6
14
24
36
50

1.3 Python映射多个函数

在以下示例中,我们将展示如何在 Python中使用map()映射多个函数。我们遍历for循环中的元素。在每个循环中,我们创建一个包含两个值的列表,这些结果是通过对传入参数应用add()和square()函数来计算的。

# 函数1
def add(x):
    return x + x

# 函数2
def square(x):
    return x * x

# 数据
nums = [1, 2, 3, 4, 5]

# 逐个取数处理
for i in nums:
    
    # lambda为处理函数,分别将add和square传给lambda
    vals = list(map(lambda x: x(i), (add, square)))

    print(vals)
[2, 1]
[4, 4]
[6, 9]
[8, 16]
[10, 25]

1.4 基于列表推导实现map函数

Python的map()也可以通过Python列表推导来实现。

# 定义函数
def square(x):

    return x * x

nums = [1, 2, 3, 4, 5]


# 列表推导式相当于每次从nums中取一个数放入square中
nums_squared = [square(num) for num in nums]

for num in nums_squared:

    print(num)
1
4
9
16
25

2 参考

猜你喜欢

转载自blog.csdn.net/LuohenYJ/article/details/118977004