使用重复的值来增加列 [关闭]

Question

关闭。 这个问题需要更加集中。 它目前不接受答案。

---

想改善这个问题吗？ 更新问题，使其仅通过编辑这篇文章来关注一个问题。

去年关闭。

改进这个问题

我有一个 Pandas 数据框，我想根据重复值的数量增加一列。 因此，当找到重复项时，所有其他出现的次数都会增加。 所以给定这个输入数据框

    SM
 0  AB
 1  AC
 2  AD
 3  AB
 4  AB
 5  AC
 6  AE
 7  AD

返回

     SM DM
  0  AB AB
  1  AC AC
  2  AD AD
  3  AB AB_1
  4  AB AB_2
  5  AC AC_1
  6  AE AE
  7  AD AD_1

我试过这行代码，但我不知道如何递增

 np.where(a.SM.duplicated(keep='first'), a.SM+'_1', a.SM)

Answer 1

使用groupby.cumcount和Series.where ：

s = df.groupby('SM').cumcount()

df['DM'] = df['SM'].where(s.eq(0), df['SM'] + '_' + s.astype(str))

[出去]

   SM    DM
0  AB    AB
1  AC    AC
2  AD    AD
3  AB  AB_1
4  AB  AB_2
5  AC  AC_1
6  AE    AE
7  AD  AD_1

Answer 2

在dplyr按组创建 counter-1 然后paste()可以为您提供想要的结果：

library(dplyr)
library(tidyr)
# Getting those whose value which Are repeated
df$BoolDup<-duplicated(df$SM)
# Creating counting variable and a second counter that keeps track of the repetitions-1 to then Join if duplicated
df %>% mutate(count = 1) %>% 
  group_by(SM)%>%
  mutate(count2 = cumsum(count)-1) %>%
  mutate(DM = ifelse(BoolDup==TRUE,paste(SM,"_",count2,sep =""), SM))%>%
  dplyr::select(SM=SM, DM=DM)

# A tibble: 8 x 2
# Groups:   SM [4]
# SM    DM   
# <chr> <chr>
# 1 AB    AB   
# 2 AC    AC   
# 3 AD    AD   
# 4 AB    AB_1 
# 5 AB    AB_2 
# 6 AC    AC_1 
# 7 AE    AE   
# 8 AD    AD_1