如何“嵌入”一个小的numpy数组到一个大numpy数组的预定义块中?

问题描述:

我有一个小的NXN数组“块”,我想要插入大型数组“墙”的指定区域(即“开始”的对角线区域)。有没有一种有效的方法来归档?如何“嵌入”一个小的numpy数组到一个大numpy数组的预定义块中?

wall[start:start+N][start:start+N] = block[:][:] 

目前我在做什么很简单:

for i in xrange(N): 
    wall[start+i][start:start+N] = block[i][:] 

则可以使用多维度指标:

import numpy as np 

wall = np.zeros((10,10),dtype=np.int) 
block = np.arange(1,7).reshape(2,3) 

x = 2 
y = 3 
wall[x:x+block.shape[0], y:y+block.shape[1]] = block 

输出为:

>>> wall 
array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 1, 2, 3, 0, 0, 0, 0], 
     [0, 0, 0, 4, 5, 6, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]) 
+0

谢谢!我花了2分钟才发现你的正确版本和我的天真之间的区别.. ;-) – nye17