python - 根据前两列值派生新列

Question

我有一个如下所示的熊猫数据框，我想导出转换我的数据框，如下所示。

逻辑：对于相同的 id，如果一个国家/地区被重复，那么我想用计数标记它。例如，对于 id 1，USA 的第一次出现，我想将其计为 1，然后当 USA 为同一用户第二次出现时，然后为 2，依此类推，如果 USA 被分配给相同的用户 id，则为第 3 次1，那么我希望计数为 3。

输入：

id  country
1   USA
1   USA
1   UK
2   USA
2   India
2   India
3   India
3   India

输出：

id  country count
1   USA 1
1   USA 2
1   UK  1
2   USA 1
2   India   1
2   India   2
3   India   1
3   India   2

标签： pythonpandas