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

如何找出 NumPy 陣列中多個值的行索引？

NumPy 提供了多種方法來查找NumPy 數組中指定值的行索引：

方法#1：NumPy廣播

方法#2：使用np.ravel_multi_index的記憶體高效方法

方法#3：記憶體高效接近與np.searchsorted

理解np.ravel_multi_index

np.ravel_multi_index將多維索引元組轉換為網格的線性索引。它假設每一列代表一個維度，並使用網格形狀來計算線性索引。

例如，使用 X:

且 dims = [10, 7]，第一行X (4, 2) 的值轉換為線性索引 30。這對應於grid:

為唯一線性索引選擇尺寸

為了確保唯一線性索引，網格的尺寸應設定為每個軸的最大拉伸輸入數組X。這可以透過將 X 中每列的最大值加 1 來實現。

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