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

我有一个小的NXN阵列“块”,我想要插入大阵列“墙”的指定区域(即“开始”的对角线区域).有没有一个有效的方法存档这个?

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]])

转载注明原文:python – 如何将一个小的numpy数组“嵌入”到大型numpy数组的预定义块中? - 代码日志