Python：使用列表索引 [from:to] 任意 numpy 数组

Question

我想将一个矩阵中的一个块复制到另一个矩阵的一个块中。 要将它与任何类型的 n 维数组一起使用，我需要通过 [] 运算符应用带有偏移量的列表。有没有办法做到这一点？

mat_bigger[0:5, 0:5, ..] = mat_smaller[2:7, 2:7, ..] 

喜欢：

off_min = [0,0,0]
off_max = [2,2,2]
for i in range(len(off_min)):
  mat_bigger[off_min[i] : off_max[i], ..] = .. 

Answer 1

您可以通过创建slice 对象的元组来做到这一点。例如：

mat_big = np.zeros((4, 5, 6))
mat_small = np.random.rand(2, 2, 2)

off_min = [2, 3, 4]
off_max = [4, 5, 6]

slices = tuple(slice(start, end) for start, end in zip(off_min, off_max))

mat_big[slices] = mat_small