我有一个如下所示的pandas数据框，我想导出变换我的数据框，如下所示。
逻辑：对于相同的ID，如果一个国家/地区重复，那么我想用计数标记它。例如，对于id为1的美国首次出现，我想将其计数为1，然后当USA对于同一用户第二次出现时，然后是2，依此类推，例如第三次将USA分配给同一用户id 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