如何正確使用Numpy的where函數進行多條件數組元素存取？

透過Numpy 的Where函數使用多個條件存取陣列元素

在NumPy 中使用陣列時，通常需要選擇性地存取基於在特定條件下。 where() 函數在這種情況下起著至關重要的作用，允許靈活選擇滿足給定條件的元素。

但是，如果 where() 的期望行為偏離預期，就會出現實例，從而導致錯誤或意想不到的結果。了解此函數的細微差別對於在 NumPy 中進行有效的陣列操作至關重要。

考慮以下程式碼片段：

<code class="python">dists[(np.where(dists >= r)) and (np.where(dists <p>此程式碼旨在選擇指定範圍內的距離 [r, r博士]。但是，它僅捕獲滿足第二個條件 dists </p>
<pre class="brush:php;toolbar:false"><code class="python">dists[abs(dists - r - dr/2.) <pre class="brush:php;toolbar:false"><code class="python">dists[(dists >= r) & (dists <p>原始程式碼中的問題源自於 where() 的錯誤使用。與布林數組不同，where() 傳回索引列表。透過 和 組合兩個索引列表會產生第二個列表，有效地覆蓋第一個條件。 </p>
<p>為了進一步清晰，請考慮以下內容：</p>
<pre class="brush:php;toolbar:false"><code class="python">a = np.where(dists >= r)
b = np.where(dists <p>a 和 b 的結果產生 b 。要獲得正確的布林數組，您需要使用按元素 & 運算符組合條件：</p>
<pre class="brush:php;toolbar:false"><code class="python">dists >= r & dists <p>布林數組可用後，您可以利用它進行數組選擇：</p>
<pre class="brush:php;toolbar:false"><code class="python">dists[dists >= r & dists </code>

以上是如何正確使用Numpy的where函數進行多條件數組元素存取？的詳細內容。更多資訊請關注PHP中文網其他相關文章！