在Python中重复行和列

时间:2016-05-31 21:00:40

标签: python numpy matrix numpy-broadcasting

我尝试制作两个600x800矩阵(XY),如下所示:
- 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的每一列,但我不确定如何走吧。

1 个答案:

答案 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]