在 groupby 特定列之后查找最早日期答案

【问题标题】：Finding earliest date after groupby a specific column在 groupby 特定列之后查找最早日期
【发布时间】：2021-11-12 06:50:33
【问题描述】：

我有一个如下所示的数据框。

  id     name   tag   location       date
   1     John    34     FL        01/12/1990
   1     Peter   32     NC        01/12/1990
   1     Dave    66     SC        11/25/1990
   1     Mary    12     CA        03/09/1990
   1     Sue     29     NY        07/10/1990
   1     Eve     89     MA        06/12/1990
   :      :      :      :              :
   n     John    34     FL        01/12/2000
   n     Peter   32     NC        01/12/2000
   n     Dave    66     SC        11/25/1999
   n     Mary    12     CA        03/09/1999
   n     Sue     29     NY        07/10/1998
   n     Eve     89     MA        06/12/1997

我需要根据 id 列找到位置信息，但有一个条件，只需要最早的日期。例如，id=1 组的最早日期是 01/12/1990，这意味着位置是 FL 和 NC。然后将其应用于所有不同的 id 组以获得前 3 个位置。我已经为我编写了代码。

#Get the earliest date base on id group
df_ear = df.loc[df.groupby('id')['date'].idxmin()]
#Count the occurancees of the location
df_ear['location'].value_counts()

代码工作得很好，但如果它们具有相同的最早日期，则它不能返回超过 1 个位置（使用我的第一行代码），例如，id=1 组将只返回 FL 而不是 FL 和 NC。我想知道如何修复我的代码以包含最早日期大于 1 的条件。

谢谢！

【问题讨论】：

标签： python pandas pandas-groupby

【解决方案1】：

将GroupBy.transform 用于系列以获得每组最少的日期，因此可以按boolean indexing 中的列Date 进行比较：

df['date'] = pd.to_datetime(df['date'])

df_ear = df[df.groupby('id')['date'].transform('min').eq(df['date'])]

【讨论】：

效果很好。非常感谢您的帮助！今天学习新事物！