python - 将交替值添加到具有重复值的字符串

Question

我有一个 pandas 数据框，在 ID 列中，值是字符串:1,1,2,2,3,3,4,4,5,5, etc...。

我想添加到字符串中，使其看起来像 1.1,1.2,2.1,2.2,3.1,3.2, etc...

理想情况下，如果有第三个或第四个相同的值，那么它会在逻辑上更新为 1.1、1.2、1.3、1.4 等...

如何最好地实现这一目标？

Answer 1

你必须使用 GroupBy.cumcount .然后使用 Series.str.cat 连接它们.

# s = pd.Series(['1', '1', '2', '2', '3', '3', '4', '4', '5', '5'])
s.str.cat(s.groupby(s).cumcount().add(1).astype(str), sep='.')

# 0    1.1
# 1    1.2
# 2    2.1
# 3    2.2
# 4    3.1
# 5    3.2
# 6    4.1
# 7    4.2
# 8    5.1
# 9    5.2
# dtype: object