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

Question

我想将一个矩阵中的一个 block 复制到另一个矩阵的一部分中。要将其与任何类型的 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