numpy 數組的條件操作

Question

我是 NumPy 的新手，在 numpy 數組上運行一些條件語句時遇到了問題。 假設我有 3 個 numpy 數組，如下所示：

A：

[[0, 4, 4, 2],
 [1, 3, 0, 2],
 [3, 2, 4, 4]]

乙：

[[6, 9, 8, 6],
 [7, 7, 9, 6],
 [8, 6, 5, 7]]

並且，c：

[[0, 0, 0, 0],
 [0, 0, 0, 0],
 [0, 0, 0, 0]]

我有一個 a 和 b 的條件語句，其中我想使用 b 的值（如果滿足 a 和 b 的條件）來計算 c 的值：

c[(a > 3) & (b > 8)]+=b*2

我收到一條錯誤消息：

Traceback (most recent call last):
  File "<interactive input>", line 1, in <module>
ValueError: non-broadcastable output operand with shape (1,) doesn't match the broadcast shape (3,4)

知道我怎么能做到這一點嗎？

我希望 c 的輸出如下所示：

[[0, 18, 0, 0],
 [0, 0, 0, 0],
 [0, 0, 0, 0]]

Answer 1

您可以使用numpy.where ：

np.where((a > 3) & (b > 8), c + b*2, c)
#array([[ 0, 18,  0,  0],
#       [ 0,  0,  0,  0],
#       [ 0,  0,  0,  0]])

或算術：

c + b*2 * ((a > 3) & (b > 8))
#array([[ 0, 18,  0,  0],
#       [ 0,  0,  0,  0],
#       [ 0,  0,  0,  0]])

Answer 2

問題是你屏蔽了接收部分，但沒有屏蔽發送部分。 其結果：

c[(a > 3) & (b > 8)]+=b*2
# ^ 1x1 matrix        ^3x4 matrix

尺寸不一樣。 鑒於您想執行逐元素添加（基於您的示例），您也可以簡單地將切片添加到右側部分：

c[(a > 3) & (b > 8)]+=b[(a > 3) & (b > 8)]*2

或者讓它更有效率：

mask = (a > 3) & (b > 8)
c[mask] += b[mask]*2

Answer 3

對 numpy 表達式稍作改動就會得到想要的結果：

c += ((a > 3) & (b > 8)) * b*2

在這里首先我創建一個帶有布爾值的掩碼矩陣，來自((a > 3) & (b > 8)) ，然后將矩陣與b*2相乘，然后生成一個3x4矩陣，可以很容易地添加到c