NumPy 的 where 函數中的多個條件
NumPy 中，where() 函數常用於條件選擇。在處理多個條件時，了解如何有效地組合它們以獲得所需的結果非常重要。
考慮一個範例，我們想要選擇指定範圍內的距離。以下程式碼嘗試執行此操作：
<code class="python">dists[(np.where(dists >= r)) and (np.where(dists <= r + dr))]
但是，這會產生意外結果，僅選擇第二個條件 (np.where(dists <= r dr)) 內的距離。
修復程式碼
要解決這個問題，我們需要了解 np.where() 傳回滿足條件的元素的索引，而不是布林數組。因此，組合多個 np.where() 呼叫的結果不會產生布林數組。
我們可以使用元素布林運算子來執行所需的條件選擇。以下是兩種正確的實作方法：
選項 1：組合條件
<code class="python">dists[(dists >= r) & (dists <= r + dr)]
& 運算子執行元素與，產生布林數組。然後我們可以用它來索引原始數組 dists。
選項2：使用中間變數
<code class="python">mask1 = dists >= r mask2 = dists <= r + dr dists[(mask1) & (mask2)]
透過為每個條件建立臨時變量，我們可以檢查兩個條件並使用& 運算子組合它們以建立布林數組。
為什麼原始程式碼不起作用
原始程式碼不起作用，因為 np .where() 傳回索引列表，而不是布林數組。組合兩個索引列表不會給出所需的結果。
例如：
<code class="python">dists = np.arange(0, 10, 0.5) r = 5 dr = 1 mask1 = np.where(dists >= r) mask2 = np.where(dists <= r + dr) print(mask1 and mask2) # Outputs: (array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]),)</code>
如您所見，結果數組不是指示哪些元素滿足這兩個條件的布林數組。
以上是如何在NumPy的where函數中有效組合多個條件？的詳細內容。更多資訊請關注PHP中文網其他相關文章！
