Python starmap()和map()应用数据

  • Post category:Python

首先,map()和starmap()都是Python内置函数,它们的作用是对可迭代对象中的元素进行指定操作(函数)后,返回一个可迭代对象。两个函数的主要区别在于,map()函数对每一个元素应用函数,而starmap()函数中的每一个元素应作为函数的单独参数传递进去。

Python的starmap()函数适用于有多个可迭代参数的情况,其中第一个参数是要执行的函数,剩下的参数是该函数的单个参数的元组。starmap()函数返回一个可迭代对象,其中的每个元素都是函数的返回值。

以下是starmap()函数的用法示例:

from itertools import starmap

def add(a, b):
    return a + b

numbers = [(1, 2), (3, 4), (5, 6)]

result = list(starmap(add, numbers))
print(result) # [3, 7, 11]

这个例子中,我们定义了一个名为add()的函数,它接受两个参数并将它们相加。我们创建了一个包含元组的列表numbers,每个元组都包含两个要加的数字。我们在starmap()中使用add()函数和numbers列表,它针对每个元组中的元素调用add()函数,并返回结果的列表。

接下来,让我们看一个map()函数的使用示例。map()函数的用法类似于starmap(),但它的不同之处在于,它在每个迭代器中都只接受一个参数,并使用该参数调用函数。

def square(x):
    return x*x

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

result = list(map(square, numbers))
print(result) # [1, 4, 9, 16, 25]

在这个例子中,我们定义了一个名为square()的函数,它接受单一参数并将其平方。接着,我们定义了一个包含数字的列表numbers,我们将square()函数和numbers列表作为参数传递给map()函数,map()函数将函数应用于数字列表中的每个元素,并返回结果的列表。

最后,需要注意的一点是,map()和starmap()函数通常会返回一个惰性迭代器对象(lazy iterator),因此需要使用list()函数将其转换为列表,以便查看其结果。