python - 如何将一个小的 numpy 数组 "embed"变成一个大的 numpy 数组的预定义 block ？

Question

我有一个小的 NXN 数组“ block ”，我想插入一个大数组“墙”的指定区域(即“开始”处的对角线区域)。有没有有效的存档方法？

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

目前我正在做的只是:

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

Answer 1

你可以使用多维索引:

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