我尝试制作两个600x800矩阵(X
和Y
),如下所示:
- X
每行有0到600个
- Y
每列有800到0
我的想法是按照以下方式做点什么:
N = np.arange(0, 600, 1)
M = np.arange(0, 800, 1)
X = np.zeros((800, 600))
Y = np.zeros((800, 600))
从此处开始,将N
放入X
的每一行,并将M[::-1]
放入Y
的每一列,但我不确定如何走吧。
答案 0 :(得分:2)
你很幸运,因为NumPy在np.meshgrid
中完全具有内置功能 -
X,Y = np.meshgrid(N,M[::-1])
示例运行 -
In [2]: N
Out[2]: array([0, 1, 2, 3, 4, 5])
In [3]: M
Out[3]: array([0, 1, 2, 3, 4, 5, 6, 7])
In [4]: X,Y = np.meshgrid(N,M[::-1])
In [5]: X
Out[5]:
array([[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5],
[0, 1, 2, 3, 4, 5]])
In [6]: Y
Out[6]:
array([[7, 7, 7, 7, 7, 7],
[6, 6, 6, 6, 6, 6],
[5, 5, 5, 5, 5, 5],
[4, 4, 4, 4, 4, 4],
[3, 3, 3, 3, 3, 3],
[2, 2, 2, 2, 2, 2],
[1, 1, 1, 1, 1, 1],
[0, 0, 0, 0, 0, 0]])
由于问题也用NumPy broadcasting
标记,我会在用零填充后(在问题中列出)添加另一种使用它的方法,就像这样 -
m,n = M.size, N.size # Size parameters
X = np.zeros((m,n),dtype=N.dtype)
Y = np.zeros((m,n),dtype=M.dtype)
X += N
Y += M[::-1,None]