31、高级NumPy使用指南

高级NumPy使用指南

1. 创建自定义NumPy通用函数（ufuncs）

在Python中创建自定义NumPy ufuncs有多种方法。最常用的是使用NumPy C API，但这里主要介绍纯Python ufuncs的创建。

• 使用 numpy.frompyfunc ：该函数接受一个Python函数以及输入和输出的数量规范。例如，创建一个逐元素相加的函数：

import numpy as np

def add_elements(x, y):
    return x + y

add_them = np.frompyfunc(add_elements, 2, 1)
result = add_them(np.arange(8), np.arange(8))
print(result)

使用 frompyfunc 创建的函数总是返回Python对象数组，这可能不太方便。

• 使用 numpy.vectorize ：该函数允许指定输出类型。

add_them = np.vectorize(add_elements, otypes=[np.float64])
result = add_them(np.arange(8), np.arange(8))
print(result