np.mgrid & np.meshgrid

mgrid

mgrid[[1:3:3j, 4:5:2j]] 
# 3j：3个点

• 步长为复数表示点数，左闭右闭
• 步长为实数表示间隔，左闭右开

X,Y = np.mgrid[1:3:3j,4:5:2j]

mgrid的作用如下图所示

形成一个3 * 2的矩阵，X,Y都是3*2的矩阵，分别是行，列坐标：

X,Y = np.mgrid[1:3:3j,4:5:2j]
print(np.mgrid[1:3:3j,4:5:2j])
print('X=',X,'\nY=',Y)

[[[1. 1.]
  [2. 2.]
  [3. 3.]]

 [[4. 5.]
  [4. 5.]
  [4. 5.]]]
X= [[1. 1.]
 [2. 2.]
 [3. 3.]] 
Y= [[4. 5.]
 [4. 5.]
 [4. 5.]]

meshgrid

meshgrid的作用如下图所示

x = np.array([1,2,3])
y = np.array([4,5])
X,Y = np.meshgrid(x,y)
print(np.meshgrid(x,y))
print('X=',X,'\nY=',Y)

[array([[1, 2, 3],
       [1, 2, 3]]), array([[4, 4, 4],
       [5, 5, 5]])]
X= [[1 2 3]
 [1 2 3]] 
Y= [[4 4 4]
 [5 5 5]]