如何從作爲目標數組的子集的源數組中分配？

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

可以簡單地獲得b的shape如：

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，...陣列。

Answer 2

獲取陣列的形狀要被分配，然後slice目標數組和分配 -

m,n = b.shape 
a[:m,:n] = b