python – 将矩阵中的每个元素重复为一个新的矩阵

参见英文答案 > Quick way to upsample numpy array by nearest neighbor tiling ????????????????????????????????????3个
在Matlab中,有一个命令repelem,其工作原理如下(见 https://www.mathworks.com/help/matlab/ref/repelem.html#buocbhj-2)：

例如：创建一个矩阵,并将每个元素重复为新矩阵的3乘2块.

A = [1 2; 3 4]

B = repelem(A,3,2)

A = (2×2)
     1     2
     3     4

B = (6×4)
     1     1     2     2
     1     1     2     2
     1     1     2     2
     3     3     4     4
     3     3     4     4
     3     3     4     4

在Numpy做同样的事情的最佳方法是什么？

A = np.arange(1,5).reshape((2,2))
B = ...

repelem = lambda a,x,y: np.repeat(np.repeat(a,axis=0),y,axis=1)
# same as repelem  = lambda a,y: a.repeat(x,0).repeat(y,1)

>>> a = np.array([[1,2],[3,4]])
>>> repelem(a,2)

array([[1,1,2,[1,4,4],4]])