numpy生成网格方法

方法1 np.meshgrid(x, y)

假设 x, y 分别为 m, n 维向量，则矩阵（数组）X, Y 的 dimension 都是： m*n 。其中矩阵 X 中的行都为向量 x，矩阵 Y 的列都为向量 y

import numpy as np
import matplotlib.pyplot as plt

m, n = 5, 4
x = np.linspace(0, m-1, m)
y = np.linspace(0, n-1, n)

# x: array([0., 1., 2., 3., 4.])
# y: array([0., 1., 2., 3.])

X, Y = np.meshgrid(x, y)
print(X)
print(Y)

输出如下

# X
array([[0., 1., 2., 3., 4.],
       [0., 1., 2., 3., 4.],
       [0., 1., 2., 3., 4.],
       [0., 1., 2., 3., 4.]])
# Y
array([[0., 0., 0., 0., 0.],
       [1., 1., 1., 1., 1.],
       [2., 2., 2., 2., 2.],
       [3., 3., 3., 3., 3.]])

X = np.reshape(X,(m * n, 1))
Y = np.reshape(Y, (m * n, 1))
arr = np.concatenate((X, Y), axis=1).reshape((m,n,-1))
print(arr)

输出

shape 5*4*2

方法2 np.tile

m, n = 5, 4
grid_x = np.tile(np.arange(n), (m, 1))
grid_y = np.tile(np.arange(m), (n, 1)).transpose()
# X: array([0., 1., 2., 3., 4.])
# y: array([0., 1., 2., 3.])
print("X")
print(grid_x)
print("Y")
print(grid_y)
grid_x = np.reshape(grid_x , (m * n, 1))
grid_y = np.reshape(grid_y , (m * n, 1))
arr = np.concatenate((grid_x, grid_y ), axis=1).reshape((m, n, -1))
print("arr")
print(arr)