如何从源数组中分配目标数组的子集？

Question

有两个数组,如:

a = np.zeros((3, 4), dtype=int)
[[0 0 0 0]
 [0 0 0 0]
 [0 0 0 0]]

b = np.ones((2, 3), dtype=int)
[[1 1 1]
 [1 1 1]]

如何从源数组(b)分配到源中存在的目标数组(a)中的条目？

结果数组应该是:

[[1 1 1 0]
 [1 1 1 0]
 [0 0 0 0]]

Answer 1

你可以简单地获得shape的b,如:

m,n = b.shape

然后使用切片设置元素a:

a[:m,:n] = b

这会产生:

>>> m,n = b.shape
>>> a[:m,:n] = b
>>> a
array([[1, 1, 1, 0],
       [1, 1, 1, 0],
       [0, 0, 0, 0]])

如果a并且b具有相同但任意数量的维度,我们可以使用以下生成器:

a[tuple(slice(mi) for mi in b.shape)] = b

这又导致:

>>> a[tuple(slice(mi) for mi in b.shape)] = b
>>> a
array([[1, 1, 1, 0],
       [1, 1, 1, 0],
       [0, 0, 0, 0]])

但这也适用于3d,4d,......阵列.