首頁 > 後端開發 > Python教學 > 如何有效率地尋找 NumPy 陣列中特定值的行索引?

如何有效率地尋找 NumPy 陣列中特定值的行索引?

Barbara Streisand
發布: 2024-12-07 17:22:13
原創
715 人瀏覽過

How to Efficiently Find Row Indices of Specific Values in a NumPy Array?

在NumPy 陣列中找出特定值的索引

問題:

給定一個數組X,確定儲存在數組X searched_values 中的指定值的行索引例如,在下面的範例中,我們尋找的索引。數組 X 中的值 [4, 2]、[3, 3] 和 [5, 6]。

所需的輸出將是:

解決方案:

方法1:NumPy >

是利用np.where的一種方式,廣播函數如下。

方法2:記憶體高效的線性索引轉換

如何使用NumPy的np.ravel_multi_index將每一行轉換為唯一的線性索引。

方法 3:記憶體高效且最佳化的線性索引轉換

NumPy 的 np.searchsorted 也可以用於快速尋找線性索引。

np.ravel_multi_index 的工作原理:

np.ravel_multi_index 建立多維索引的線性索引表示。將每一行解釋為 n 維多維數組的索引並產生相應的線性索引。

例如,將 np.ravel_multi_index(X.T, dims) 應用於樣本數組 X:

這表示數組 X 的每一行具有的線性索引。此線性索引可用於唯一標識數組中的每一行。

每個方法對應的範例程式碼如下。

您可以使用任一方法輕鬆確定數組 X 中特定值的行索引。

以上是如何有效率地尋找 NumPy 陣列中特定值的行索引?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板