問題:
給定一個數組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中文網其他相關文章!