如何在 NumPy 数组上映射函数?
在本文中,我们将看到如何在 Python 中将函数映射到 NumPy 数组上。
方法 1: numpy .矢量化()方法
numpy.vectorize()函数将函数映射到包含一系列对象(如 numpy 数组)的数据结构上。对象或 NumPy 数组的嵌套序列作为输入,并返回单个 NumPy 数组或 NumPy 数组的元组。
Python 3
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
def addTwo(i):
return i+2
applyall = np.vectorize(addTwo)
res = applyall(arr)
print(res)
版权属于:月萌API www.moonapi.com,转载请注明出处