データフレームの上位n項目に.aggを適用するには

Question

大きなデータセットの上位4項目から計算されたデータフレームオブジェクトを取り戻そうとしています。これを行うには、私はその後、作成したデータフレームcensus_dfを引っ張っそれらの状態の内部四大郡によって推定彼らの人口を持つ4つの最大の郡ではなく、すべての状態を返し、次の

return (census_df.set_index('STNAME','COUNTY') 
      .nlargest(4,'CENSUS2010POP') 
      .groupby(level=0)['CENSUS2010POP'] 
      .agg([{'Sum':np.sum}])) 

を書きました。私は

return (census_df.set_index('STNAME','COUNTY') 
      .nlargest(4,'CENSUS2010POP') 
      .groupby(level=0)['CENSUS2010POP'] 
      .agg([{'Sum':np.argsort()[-3:].sum()}])) 

が、argsortを使用してみました は位置引数が不足していると私はそこに入れて何かわからない - argsortも、何を参照するのでしょうか？完全に働いた

Answer 1

IIUC

census_df.set_index('STNAME','COUNTY').\ 
     groupby(level=0)['CENSUS2010POP'].\ 
     agg(lambda x: x.head(4).sum())