如何創建一個包含多變量函數值的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) 

Answer 2

xyz = numpy.mgrid[-5:5,-5:5,-5:5] 
sinxyz = numpy.sin(xyz[0]+xyz[1]+xyz[2]) 

Answer 3

的numpy.mgrid功能將發揮同樣的

x,y,z = numpy.mgrid[x_min:x_max:x_num, y_min:y_max:y_num, z_min:z_max:z_num] 
sinxyz = numpy.sin(x+y+z) 

編輯：以得到它的工作x_num，y_num和z_num必須是顯式數字，後跟j，例如，x,y = numpy.mgrid[-1:1:10j, -1:1:10j]