python - 如何使用 numpy 数组评估 Python 中的函数？

Question

我有两个随机生成的 x 和 y 数组，我想在某个函数上评估每一对。我如何使用 numpy 做到这一点？我尝试了下面的代码，但它不起作用。

我想要的是一个 func(x_i,y_i) 数组。

x = np.random.uniform(0, 1, 100)
y = np.random.uniform(0, 1, 100)

func = np.array((math.exp(-2 * x) * math.cos(2 * y)))

有任何想法吗？

标签： pythonnumpy

Answer 1

不使用math，使用numpy。

import numpy as np

x = np.random.uniform(0, 1, 100)
y = np.random.uniform(0, 1, 100)

func = np.exp(-2 * x) * np.cos(2 * y)

>>> func.shape
    (100,)

也就是说，func[i]包含使用x[i]和y[i]作为输入的函数的结果。