python - 使用相同的键对过去的条目进行计数

Question

我有这个简化的数据集，其中包含日期和 ID。我想要一个包含 ID 计数的新列总计，但由于我正在创建 ML 模型预测，因此我只想考虑实际事件之前的条目。

我想只计算日期较小的每个条目。类似于:计算 ID 相同且日期 <= 当前行日期的所有条目，并添加具有该值的列。

ID | Date   | 
1  | 3/4/10 |
1  | 3/6/10 |
1  | 1/2/10 |
2  | 5/5/10 |
2  | 5/6/10 |

结果

ID | Date   | Total
1  | 3/4/10 | 1
1  | 3/6/10 | 2
1  | 1/2/10 | 0
2  | 5/5/10 | 0
2  | 5/6/10 | 1

非常感谢!

Answer 1

这是一个简化的伪代码，可以使用以下任一方法实现itertools 或 pandas:

• 按 ID 列对数据集进行分组
• 对于每个组:
  • 对日期列进行排序
  • Total 列是日期的序号。