如何根據其他兩個 numpy arrays 僅使用 Z2EA9510C37F7F821ECBFZF24 操作的條件獲得新的 numpy 陣列？

Question

假設我有a = [0, 1, 1, 0, 0, 1]和b = [1, 1, 0, 0, 0, 1]的 np.array

我想要一個新矩陣 c 使得如果a[i] = 0和b[i] = 0那么c[i] = True

我試過使用c = np.where(a == 0 & b==0)但似乎這不適用於兩個 arrays。

最終，我想要在不使用任何循環和外部庫的情況下計算兩種情況下都為真的'0'和兩種情況下都為真的'1'。 請指教。

Answer 1

這是我的解決方案：

您的邏輯表達式不過是 c = NOT (a OR b)。 這意味着，只有當a[i]和b[i]都為零時，您才需要c[i] = True 。 OR 可以通過添加兩個 arrays 來實現。 然后我們將數組轉換為 boolean 類型並將其反轉。

import numpy as np

a = np.array([0, 1, 1, 0, 0, 1])
b = np.array([1, 1, 0, 0, 0, 1])

c = np.invert((a+b).astype('bool'))

如果你想計算零的數量，你可以簡單地執行

n_zeros = np.sum(c)

更通用的解決方案：

如果你想要你的數組c[i] = True if a[i] == a0 and b[i] == b0 ，你可以這樣做：

c = (a == a0) & (b == b0)

條件a == a0和b == b0分別返回一個 boolean 數組，其中包含每個單獨數組元素的條件真值。 按位運算符&執行按元素邏輯與。 有關更多按位運算符，請參閱https://wiki.python.org/moin/BitwiseOperators 。