[解決済み] NumPyの配列で、N個の最大値のインデックスを得るには?
2022-02-14 07:54:15
質問内容
NumPyでは、配列の最大値のインデックスを取得する方法として、以下の方法を提案しています。
np.argmax
.
同じようなことをしたいのですが、その場合は
N
最大値です。
例えば、配列があったとして
[1, 3, 2, 4, 5]
,
function(array, n=3)
はインデックスを返します。
[4, 3, 1]
という要素に対応する
[5, 4, 3]
.
解決方法は?
私が思いついた最もシンプルなものは
In [1]: import numpy as np
In [2]: arr = np.array([1, 3, 2, 4, 5])
In [3]: arr.argsort()[-3:][::-1]
Out[3]: array([4, 3, 1])
これには、配列の完全なソートが必要です。もし
numpy
は、部分的なソートを行うための組み込みの方法を提供します; これまでのところ、私はそれを見つけることができませんでした。
この解決策が遅すぎることが判明した場合(特に小さな
n
でコーディングすることを検討する価値があるかもしれません。
Cython
.
関連
-
PythonによるLeNetネットワークモデルの学習と予測
-
[解決済み] Pythonで現在時刻を取得する方法
-
[解決済み] 列の値に基づいてDataFrameから行を選択するにはどうすればよいですか?
-
[解決済み] 環境変数の値にアクセスする方法
-
[解決済み] Python 3で「1000000000000000 in range(1000000000000001)」はなぜ速いのですか?
-
[解決済み] リストの最後の要素を取得する方法
-
[解決済み] Pythonで文字列の部分文字列を取得するにはどうすればよいですか?
-
[解決済み] モジュールの関数名(文字列)を使って、モジュールの関数を呼び出す。
-
[解決済み] NumPyの配列で0に等しい要素のインデックスを探す
-
[解決済み] NumPyの配列の中で、1つの軸に沿った最大要素のインデックスを取得する方法
最新
-
nginxです。[emerg] 0.0.0.0:80 への bind() に失敗しました (98: アドレスは既に使用中です)
-
htmlページでギリシャ文字を使うには
-
ピュアhtml+cssでの要素読み込み効果
-
純粋なhtml + cssで五輪を実現するサンプルコード
-
ナビゲーションバー・ドロップダウンメニューのHTML+CSSサンプルコード
-
タイピング効果を実現するピュアhtml+css
-
htmlの選択ボックスのプレースホルダー作成に関する質問
-
html css3 伸縮しない 画像表示効果
-
トップナビゲーションバーメニュー作成用HTML+CSS
-
html+css 実装 サイバーパンク風ボタン
おすすめ
-
pythonを使ったオフィス自動化コード例
-
pythonサイクルタスクスケジューリングツール スケジュール詳解
-
Python百行で韓服サークルの画像クロールを実現する
-
PyQt5はユーザーログインGUIインターフェースとログイン後のジャンプを実装しています。
-
Pythonの画像ファイル処理用ライブラリ「Pillow」(グラフィックの詳細)
-
[解決済み】Pythonスクリプトで「Expected 2D array, got 1D array instead: 」というエラーが発生?
-
[解決済み】Python elifの構文が無効です【終了しました
-
[解決済み】Python Error: "ValueError: need more than 1 value to unpack" (バリューエラー:解凍に1つ以上の値が必要です
-
[解決済み】Python: SyntaxError: キーワードは式になり得ない
-
[解決済み] numpyのflatten関数とravel関数の違いは何ですか?