如何有效率地找出 NumPy 數組中多個值的行索引？

在 NumPy 陣列中尋找多個值的行索引

給定一個陣列 X 和一組目標行 searched_values，任務是擷取對應的行索引。使用各種 NumPy 函數可以有效地解決這個問題。

方法1：廣播

使用廣播的簡單方法：

np.where((X==searched_values[:,None]).all(-1))[1]

方法2：記憶體高效轉換

為了記憶體效率，將每一行轉換為唯一的線性索引並使用np.in1d:

dims = X.max(0)+1
out = np.where(np.in1d(np.ravel_multi_index(X.T,dims),\
                    np.ravel_multi_index(searched_values.T,dims)))[0]

方法3：記憶體高效搜尋

使用np.searchsorted 的另一個記憶體高效解決方案：

dims = X.max(0)+1
X1D = np.ravel_multi_index(X.T,dims)
searched_valuesID = np.ravel_multi_index(searched_values.T,dims)
sidx = X1D.argsort()
out = sidx[np.searchsorted(X1D,searched_valuesID,sorter=sidx)]

注意： 這種方法假設searched_values 中的每一行都與X.

以上是如何有效率地找出 NumPy 數組中多個值的行索引？的詳細內容。更多資訊請關注PHP中文網其他相關文章！