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)

如果要实现和以上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)

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

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

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

for num in nums_squared:

    print(num)

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)

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)

巴特西

[python] Python map函数总结

Python map函数总结

1 使用

1.1 基础示例

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

1.3 Python映射多个函数

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

2 参考

最新文章

热门文章