python - 基于其他数组的掩码数组？

Question

我有一个 numpy 数组 arrayBig

10  5 27 30 34
2  34 23  2  3 
2  3  43 12 23
2  24 34  2 34

我有一个 numpy 数组 arraySmall

1 0 0 
0 1 1
1 1 0

我想要的是一个numpy数组arrayNew

34 0 0 
0 43 12
24 34 0

我知道我的 arraySmall 的形状为 (3,3) 并且位于 arrayBig 中的索引 (1 1) 。如何使用 Numpy 获取 arrayNew？

Answer 1

>>> import numpy as np
>>> arrayBig = np.array([
...     [10,  5, 27, 30, 34],
...     [2,  34, 23,  2,  3],
...     [2,   3, 43, 12, 23],
...     [2,  24, 34,  2, 34],
... ])
>>> arraySmall = np.array([
...     [1, 0, 0],
...     [0, 1, 1],
...     [1, 1, 0],
... ])
>>> arrayBig[1:4, 1:4] * arraySmall
array([[34,  0,  0],
       [ 0, 43, 12],
       [24, 34,  0]])