Получение значения столбца, где другой столбец является минимальным из группы

Question

INPUT

У меня есть входной фрейм данных с текстом, длиной символа и значением 'x'.

    x   text    len     flag
0   1   hi      2       1
1   1   hello   5       0
2   1   how     3       1
3   2   are     3       1
4   2   you?    4       1
5   2   kiddo   5       1

Я хочу сгруппировать x и получить текст с наименьшей длиной символа, где флаг равен 1.

OUTPUT

    x   text    len     select
0   1   hi      2       hi
1   1   hello   5       nan
2   1   how     3       nan
3   2   are     3       are
4   2   you?    4       nan
5   2   kiddo   5       nan

Я попытался сгруппировать и получитьминимальное значение, но как мне получить текст, где присутствует минимальное значение?Я не хочу использовать apply.

df[df.flag == 1].groupby('x')['len'].transform('min') 

Я получу минимальное значение длины символа, но не текстовое значение, где длина символа минимальная

Answer 1

Вам понадобится groupby и transform с idxmin:

# Handle the `flag` requirement.
v = df.assign(len=df['len'].where(df.flag.astype(bool)))

condition = v.groupby('x')['len'].transform('idxmin')
df.loc[condition, 'select'] = df.loc[condition, 'text'].values

df
   x   text  len  flag select
0  1     hi    2     1     hi
1  1  hello    5     0    NaN
2  1    how    3     1    NaN
3  2    are    3     1    are
4  2   you?    4     1    NaN
5  2  kiddo    5     1    NaN