如何使用numpy数组在Python中评估函数

Question

我有两个随机生成的x和y的数组，我想在某个函数中计算每一对。我该如何使用numpy？我尝试了下面的代码，但它不起作用。

我想要的是函数(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)))

有什么想法吗？

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]作为输入的函数的结果。