如何找出 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中文網其他相關文章!