グループごとに最大値を有する行を取得して、新しい列を作成しマーキングする方法(python,pandas)

Question

グループごとに最大値を有する行を取得して、新しい列にマーキングする方法

下記の表のように、kigouとpointが与えられたとします。
同一のkigouにおいてpointが高い行{(AA,40)と(CC,30)}に対して、新しい列("max_")を作成し'True'と記載したいです。

pandasのgroupby使って最大値の行は出力することが出来たのですが、表のように新しい列を作成しマーキングすることが出来ません。どなたか方法をお教え頂けないでしょうか。

df2 = df.groupby('kigou')['point'].max()

kigou	point	max_
AA	20	NaN
AA	40	True
CC	20	NaN
CC	30	True

Answer 1

以下でできます。

1. df.groupby("kigou"): 列 "kigou" でグループ化

DataFrame -> DataFrameGroupBy
2. df.groupby("kigou")["point"]: 列 "kigou" でグループ化された DataFrame の列 "point" を取得
DataFrameGroupBy -> SeriesGroupBy
3. df.groupby("kigou")["point"].idxmax(): グループごとの最大値のインデックス
SeriesGroupBy.idxmax()

import pandas as pd

df = pd.DataFrame(
    {
        "kigou": ["AA", "AA", "CC", "CC"],
        "point": [20, 40, 20, 30]
    }
)

max_idx = df.groupby("kigou")["point"].idxmax()
df.loc[max_idx, "max_"] = True
print(df)

	kigou	point	max_
0	AA	20	nan
1	AA	40	1
2	CC	20	nan
3	CC	30	1

Comments

[usuke]

ありがとうございます、おかげさまで出来ました。