問題タブ [argmax]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
149 問題
0
投票する
2
に答える
5706
参照
python - pandas idxmax: 同数の場合はすべての行を返す
私は、確率によって各行に重みを付けるデータフレームを使用しています。ここで、確率が最も高い行を選択したいので、pandas idxmax() を使用していますが、同点の場合は、同点の最初の行を返すだけです。私の場合、結ぶすべての行を取得したいと思います。
さらに、以下のような数百万のデータフレームを処理する研究プロジェクトの一環としてこれを行っているため、高速に保つことが問題です。
例：
私のデータは次のようになります。
このリストから、次のように pandas データフレームを作成します。
次のようになります。
次に、以下を使用して argmax(probability) に適合する行を選択します。
もちろん、どちらが返されますか:
同点の場合にすべての値を取得する (高速) 方法はありますか?
ありがとう！