如何使用 numpy ZA3CBC3F9D0CE2F2C1554E1BD671 评估 Python 中的 function

Question

I have two arrays of randomly generated x's and y's and I want to evaluate each pair at some function. How do I do that using numpy? I tried the code below, but it does not work.

What I want is an array of 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)))

Any ideas?

Answer 1

Don't use math , use 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,)

That is, func[i] contains the result of your function using x[i] and y[i] as input.