文章目录
NumPy 中的自定义通用函数(ufunc):为数组元素量身定制的操作
通用函数:对数组每个元素的高效操作
NumPy 的通用函数(ufunc)是对数组中每个元素执行相同操作的函数。它们是高度优化的,因此比普通 Python 函数快得多。
创建自定义 ufunc
我们可以使用 NumPy 提供的 np.frompyfunc
方法来创建自定义的 ufunc。这允许我们将普通的 Python 函数转换成 ufunc,从而在 NumPy 数组上以矢量化方式运行。
示例:自定义一个平方函数
假设我们想要一个函数来计算数组每个元素的平方。我们可以这样创建一个 ufunc:
def square(x):
return x * x