在 pandas 中查找 groupby 的索引

Posted 2023-03-12

【中文标题】在 pandas 中查找 groupby 的索引

【英文标题】：Find indices of a groupby in pandas

【发布时间】：2021-05-09 16:50:01

【问题描述】：

我有这个假数据集：

我想知道 BirthDate 和 ZipCode 的组合出现了多少次，如下所示：

现在，我的问题是如何找到数据集中发生这些事件的位置df？例如，如何找到 2000101 和 08002 的位置是？

提前致谢。

【问题讨论】：

预期输出应该如何？

df.groupby('BirthDate', 'ZipCode').apply(lambda x: list(x.index))怎么样

【参考方案1】：

如果需要聚合计数和索引值（如列表到新列 Pos），请使用 GroupBy.agg：

df1 = (df.reset_index()
         .groupby(['BithDate','ZipCode'])
         .agg(RowNumber=('BithDate','size'), Pos = ('index',list)))