pandas、ランキング作成方法

Question

pythonのpandasでのデータフレームの扱いについての質問です。

以下、例を用いて説明させて頂きます。

nameという列には果物の名前が無作為に3種類(りんご、バナナ、ぶどう)入っています。
それぞれ果物の数は異なります。

weightという列には果物の重さが入っています。種類が同じでもそれぞれの重さは異なります。

placeという列にはその果物が採れた場所(便宜上、アルファベットで表します)が入っています。

この時、果物の種類別に重さの上位５位が採れた場所を示す添付画像のようなデータフレームを作成するとしたらどのようなコードを書けば良いでしょうか。

よろしくお願いします。

Answer 1

df.groupby('name').apply(lambda d:d.sort_values(by='weight', ascending=False).iloc[:5, d.columns.get_loc('place')].reset_index(drop=True))

でどうでしょう

import pandas as pd
import numpy as np

N = 100
df = pd.DataFrame(
    {'name': np.random.choice(['りんご','バナナ','ぶどう'], N),
     'weight': np.random.uniform(0,20,N),
     'place': np.random.choice(['A','B','C','D','E','F'], N)})

ret = df.groupby('name').apply(lambda d:d.sort_values(by='weight', ascending=False).iloc[:5, d.columns.get_loc('place')].reset_index(drop=True))       print(ret)
place  0  1  2  3  4
name
ぶどう    C  B  F  A  A
りんご    C  B  F  A  C
バナナ    E  C  A  F  F