📅  最后修改于: 2023-12-03 15:25:16.530000             🧑  作者: Mango
当我们需要对向量中的每个元素应用同一个函数时,我们可以使用 map
函数。这个函数会将我们提供的函数应用于向量中的每个元素,并将结果保存到一个新的向量中。
以下是一个简单的例子,我们将一个向量 [1, 2, 3]
中的每个元素平方,得到新的向量 [1, 4, 9]
:
def square(x):
return x ** 2
r = [1, 2, 3]
result = list(map(square, r))
print(result)
输出:
[1, 4, 9]
我们将 square
函数应用于向量 r
中的每个元素,并将结果保存到新的向量 result
中。此时我们可以发现,result
中的元素是向量 r
中对应元素的平方。
除了上面示例中的自定义函数,我们还可以使用 lambda 函数来进行快速的映射:
r = [1, 2, 3]
result = list(map(lambda x: x ** 2, r))
print(result)
输出:
[1, 4, 9]
我们使用 lambda 函数定义了一个简单的平方函数,并将其传递给 map
函数来对向量 r
中的每个元素进行平方操作。
不仅如此,Python 中还提供了另外一种更简洁的方式来进行向量元素映射,那就是使用列表生成式:
r = [1, 2, 3]
result = [x ** 2 for x in r]
print(result)
输出:
[1, 4, 9]
以上就是使用 Python 将函数应用于所有向量元素的几种简便方法。无论我们使用哪种方式,都能快速、简单地对向量中的每个元素进行处理,从而获得新的向量。