集計後に一定のサンプル数に満たない条件の列をあらかじめ対象から削除したい

Question

質問内容

以下のようなdfにおいて、集計したときにサンプル数（列数）が40以下になる"Player"（を含む列）に関してはdfから除外するにはどうしたらよいでしょうか？

ご教授の程、よろしくお願い致します。

該当のソースコード

import pandas as pd
import numpy as np
pd.options.display.float_format="{:.1f}".format
#Create a random data frame
start,end = "2021/4/1","2021/4/30"
dates = pd.date_range(start=start,end=end,freq="D")
players = [f"Player{i}"for i in range(1,20)]
N = 1000
dates = np.random.choice(dates,size=N)
dates.sort()
players=np.random.choice(players,size=N)
df = pd.DataFrame({
        "Date": dates,
      "Player": players,
       "Speed": np.random.sample(N) * 100.0,
       "Angle": np.random.sample(N) * 40.0 - 20.0,
})
#resampling
dfx = df.groupby("Player").sample(n=40)

#grouping
dfy = dfx.groupby("Player").agg({
               "Player":"count",
     "Speed":[np.mean],
   "Angle":[np.mean],
})
print(dfy)

達成したいこと/発生している問題

最終的に、Playerでグルーピングした後に、.agg()にてPlayerごとの各項目の平均値を算出したいと思っています。
サンプル数がPlayerごとに偏りがあるため、算出値の一貫性を担保するためにランダムに40列ずつを抽出したいと考えておりますが、
元からサンプル数が40に満たないPlayerに関しては、
対象から除外したのちにグルーピングしなければと思っております。

#extract random 100 data from each players
dfx = df.groupby("Player").sample(n=40)
#grouping
dfy = dfx.groupby("Player").agg({
               "Player":"count",
     "Speed":[np.mean],
   "Angle":[np.mean],
})

ただし、サンプル数が40に満たないPlayerが存在すると以下のようにエラーが発生します。
あくまで重複は無しで、40に満たないPlayerに関してはあらかじめグルーピングから外す必要があると考えています。
ValueError: Cannot take a larger sample than population when 'replace=False'

補足情報（FW/ツールのバージョンなど）

windows11,python3.9.4,vscode

Answer 1

filter を使う場合です。

# dfx = df.groupby("Player").sample(n=40)
dfx = df.groupby("Player").filter(lambda x: x['Player'].count()>=40).sample(n=40)

8960さんのコメント

↑上記は全体でn=40になってしまったので、

dfx = df.groupby("Player").filter(lambda x: x['Player'].count()>=40)
dfy = dfx.groupby("Player").sample(n=40)

> と修正したところできました。

Comments

[8960]

dfx = df.groupby("Player").filter(lambda x: x['Player'].count()>=40).sample(n=40)
↑上記は全体でn=40になってしまったので、

dfx = df.groupby("Player").filter(lambda x: x['Player'].count()>=40)
dfy=dfx.groupby("Player").sample(n=40)

と修正したところできました。
助かりました。ありがとうございました。

[melian]

あ、ごめんなさい。ご指摘いただき、どうもありがとうございます。