如何“嵌入”一个小的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]])
谢谢!我花了2分钟才发现你的正确版本和我的天真之间的区别.. ;-) – nye17