熊猫 groupby 的最小和最大行

Question

我有一个 Pandas 数据框，我将它分组在“groupID”列上。

gb = df.groupby('groupID')

每一行都有 ax 和 y 坐标以及一个残差（距线 x=y 的距离）。 现在我想在残差最大和最小的组中找到 2 个点（x，y）之间的梯度。 我知道如何使用 gb['residual'].min() 和 max() 但这并没有给我行。

我该如何计算？

df[gb ['residual'].idxmin()]

Answer 1

我认为您需要通过DataFrameGroupBy.idxmax和DataFrameGroupBy.idxmin从列residual的最大值和最小值中找到所有索引，然后按loc选择：

df1 = df.loc[df.groupby('groupID').residual.idxmax()]
df2 = df.loc[df.groupby('groupID').residual.idxmin()]