首頁 > 後端開發 > Python教學 > 如何有效率地找出 NumPy 數組中多個值的行索引?

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

Barbara Streisand
發布: 2024-12-12 18:44:11
原創
460 人瀏覽過

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

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

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