【Python】２つの二次元配列を結合ではなく、重ねる方法がわからない

【Pytyon】２つの二次元配列を結合ではなく、重ねる方法がわからない

１つの二次元配列の真ん中に，もう１つの二次元配列を重ねたいのですが，その方法がわかりません．

py
1import numpy as np
2
3array1 = np.zeros((10, 10), dtype="int32")
4array2 = np.ones((4, 8), dtype="int32")
5
6array3 = overlap_array(array1, array2)
7
8def overlap_array(array1, array2):
9    # ここに重ねる処理を書きたい
10    return array3
11
12print(array3)

出力

[[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 1 1 1 1 1 1 1 1 0]
[0 1 1 1 1 1 1 1 1 0]
[0 1 1 1 1 1 1 1 1 0]
[0 1 1 1 1 1 1 1 1 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]]

上記のコード，array1の上にarray2を重ねたいのですが，どうすればいいのかがわかりません．
次の条件下でも正しくプログラムを教えて頂きたいです．

条件

array1に重ねる際，ちょうど真ん中に重ねられるようにしたい
array2の各次元のサイズが変更されても正しく動作させたい
- 6x2
```
py
1array2 = np.ones((6, 2), dtype="int32")
```
  出力
  
  [[0 0 0 0 0 0 0 0 0 0]
  [0 0 0 0 0 0 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 1 1 0 0 0 0]
  [0 0 0 0 0 0 0 0 0 0]
  [0 0 0 0 0 0 0 0 0 0]]
- 6x10
```
array2 = np.ones((8, 10), dtype="int32")
```
  出力
  
  [[0 0 0 0 0 0 0 0 0 0]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [1 1 1 1 1 1 1 1 1 1]
  [0 0 0 0 0 0 0 0 0 0]]
※ 各次元のサイズは2以上，10以下の偶数

以上，よろしくお願い致します．

行動規範の内容に同意します

回答1件

ベストアンサー

python
1import numpy as np
2import copy
3
4def overlap_array(array1, array2):
5    w1, h1 = array1.shape
6    w2, h2 = array2.shape
7    w, h = (w1-w2)//2, (h1-h2)//2
8    array3 = copy.deepcopy(array1)
9    array3[w:(-w if w else None), h:(-h if h else None)] = array2
10    return array3
11
12if __name__ == '__main__':
13    array1 = np.zeros((10, 10), dtype="int32")
14    array2 = np.ones((4, 8), dtype="int32")
15    array3 = overlap_array(array1, array2)
16
17    print(array3)