python - 如何计算 python 中的外部函数？

Question

如果我有一个像 func(x,y) = cos(x) + sen(y) + x*y 这样的随机函数，我如何将它应用于 2 个数组中的所有元素对？

我找到了 https://docs.scipy.org/doc/numpy/reference/generated/numpy.outer.html和发现所有基本操作都有外部函数。但是如果我想用自定义函数来实现呢？

假设 array1 是 [1,2]，array2 是 [3,4]，我想应用的函数叫做 f(float, float)

预期的输出是

[f(1,3) f(1,4)

f(2,3) f(2,4)]

Answer 1

只要确保以正确广播的方式编写函数，就可以做到

func(x_arr[:, None], y_arr)

将其应用于两个一维数组 x_arr 和 y_arr 中的所有元素对。

例如，要以广播的方式编写示例函数，您可以将其编写为

def func(x, y):
    return np.cos(x) + np.sin(y) + x*y

因为 np.cos、np.sin、+ 和 * 跨数组广播和向量化。

---

至于不播呢？好吧，有些人可能会建议使用 np.vectorize，但是您必须牢记很多棘手的事情，例如保持一致的输出 dtype 并且没有副作用。如果您的函数不广播，我建议您只使用列表理解:

np.array([[func(xval, yval) for yval in y_arr] for xval in x_arr])