【Pandas】 DataFrame のある列の最大値を含む行のインデックス値を取得する方法。

今回は、Pandas の DataFrame において、ある列で最大値を求めて、その最大値をもつ行についてインデックスを取得する方法について紹介します。

文章だと少し分かりずらいので、実際のデータでみていきます。

以下のような DataFrame（df）があるとします。

インデックスが「0.01」「0.02」… と連続していて、実際には「1.00」まで続きます。

カラム名が「precision」「recall」「f-measure」の３つになります。

列ごとの最大値を求める

「precision」「recall」「f-measure」カラムのそれぞれの列ごとの最大値を求めるには、「max」関数を使います。

結果は Series で返ります。これだと、行はバラバラのデータになります。

ある列の最大値を持つインデックスを求める

ある列で最大値を持っている行を抽出したいときがあると思います。

今回は、「f-measure」列の中で最大値をもつ行のインデックスの値を求めてみます。

「idxmax」関数を使います。

ある列の最大値を持つ行を抽出する（Series型）

先ほどは、行のインデックス値のみを抽出しましたが、行全体を抽出してみます。

先ほど抽出したインデックス値を loc を使って渡してあげれば行の情報が抽出できます。

Series 型で抽出されます。

ある列の最大値を持つ行を抽出する（DataFrame型）

Series 型ではなく DataFrame 型で取得したいときは、loc に [ ] を１つ追加して、リストで渡します。

無事 DataFrame 型で取得できました。