如何创建一个包含多变量函数值的numpy数组？

Question

我想创建一个包含函数的数组f(x,y,z).如果它是我要做的一个变量的函数,例如:

sinx = numpy.sin(numpy.linspace(-5,5,100))

获得sin(x)了x在[-5,5]

例如,我怎么能这样做sin(x+y+z)呢？

Answer 1

我似乎找到了一种方法:

# define the range of x,y,z
x_range = numpy.linspace(x_min,x_max,x_num)
y_range = numpy.linspace(y_min,y_max,y_num)
z_range = numpy.linspace(z_min,z_max,z_num)

# create arrays x,y,z in the correct dimensions
# so that they create the grid
x,y,z = numpy.ix_(x_range,y_range,z_range)

# calculate the function of x, y and z
sinxyz = numpy.sin(x+y+z)