获取按数据框分组的多索引上具有最大列值的索引

Question

我正在处理数据集上的熊猫，我想获得在比赛中取得更多胜利的属性。 我能够使用 groupby 函数创建一个数据框。 例如，对于可能有 3+ 个替代项的属性“surface”，我有这个数据框：

现在我想要一个输出数据框，如：

fullname              best_surface
Zuzana Zlochova       Hard
Zuzanna Bednarz       Clay
....

我设法通过对只能具有两个值的属性进行一些合并来解决此问题，但它不适用于可以具有 3 个或更多值的属性。 数据集很大，所以我必须使用 Pandas 操作，我不能使用 iters。

谢谢

Answer 1

使用DataFrameGroupBy.idxmax为指数由第一最大柱hasWon ，选择行和转换MultiIndex ，以DataFrame由MultiIndex.to_frame ：

df = df.loc[df.groupby(level='fullname')['hasWon'].idxmax()].index.to_frame(index=False)
print (df)
              fullname surface
0      Zuzana Zlochova    Hard
1      Zuzanna Bednarz    Clay
2  Zuzanna Szczepanska    Clay
3   Zvonimir Oreskovic    Hard

或者在构造函数DataFrame元组转换为DataFrame ：

df = pd.DataFrame(df.groupby('fullname')['hasWon'].idxmax().tolist(), 
                  columns=['fullname','best_surface'])
print (df)
              fullname best_surface
0      Zuzana Zlochova         Hard
1      Zuzanna Bednarz         Clay
2  Zuzanna Szczepanska         Clay
3   Zvonimir Oreskovic         Hard