我正在尝试编写一个函数,在2d数组中创建一个5像素乘5像素的规则网格.我希望numpy.arange和numpy.repeat的某些组合可能会这样做,但到目前为止我还没有运气,因为numpy.repeat只会沿着同一行重复.
这是一个例子：
假设我想要一个2d阵列形状(20,15)内的5×5网格.它应该看起来像：
array([[ 0,1,2,2],[ 0,[ 3,3,4,5,5],[ 6,6,7,8,8],[ 9,9,10,11,11],11]])
我意识到我可以简单地使用循环和切片来实现这一点,但我可以将它应用于非常大的数组,我担心它的性能太慢或不切实际.
有人能推荐一种方法来实现这一目标吗？
提前致谢.
更新：
提供的所有答案似乎都运作良好.任何人都可以告诉我哪些对大型阵列使用效率最高？通过大阵列,我的意思是它可以是100000 x 100000或更多,具有15 x 15网格单元尺寸.
解决方法
广播是这里的答案：
m,n,d = 20,15,5 arr = np.empty((m,n),dtype=np.int) arr_view = arr.reshape(m // d,d,n // d,d) vals = np.arange(m // d * n // d).reshape(m // d,1) arr_view[:] = vals >>> arr array([[ 0,11]])