在R中重置爲1的組編號爲

Question

我確信這必須是一個愚蠢的但我不能得到它的工作。我想添加一個ID col到一個數據框，爲一列中的每個唯一值重置爲1。描述最好的方法是通過例如：

gr1 <- c("A","A","A","B","B","B") 
gr2 <- c(1,1,2,3,4,4) 

df <- data.frame(gr1, gr2) 

希望的輸出：

id <- c(1,1,2,1,2,2) 
df <- cbind(df, id) 

的ID被標記GR1的每個子集內的GR2唯一值。當gr1從A更改爲B時，ID重置爲1.我已閱讀（Assign an ID based on two columns R），但這不是我想要的。我不想添加排名功能（我認爲），因爲我希望我的領帶在gr1中都具有相同的ID，例如

df2 <- df %>% group_by(gr1) %>% mutate(id=rank(gr2, ties.method="max")) 

把我的頭撞在牆上。任何指針都會有很大的幫助。

Answer 1

我們可以使用?rleid從data.table包。

library(data.table) 
setDT(df)[, id := rleid(gr2), by = gr1] 
> df 
    gr1 gr2 id 
1: A 1 1 
2: A 1 1 
3: A 2 2 
4: B 3 1 
5: B 4 2 
6: B 4 2 

Answer 2

試試這個，它使用ave進行分組和factor從1.注意ave自動由於GR2是數字，它使結果一致的因素轉換回數字重新分配順序的水平。沒有包被使用。

df2 <- transform(df, gr2 = ave(gr2, gr1, FUN = factor)) 

，並提供：

> df2 
    gr1 gr2 
1 A 1 
2 A 1 
3 A 2 
4 B 1 
5 B 2 
6 B 2 

它返回的因素和數字列的數據幀：

> str(df2) 
'data.frame': 6 obs. of 2 variables: 
$ gr1: Factor w/ 2 levels "A","B": 1 1 1 2 2 2 
$ gr2: num 1 1 2 1 2 2 

Answer 3

這裏是一個dplyr解決

df %>% 
    group_by(gr1) %>% 
    mutate(id=as.numeric(factor(gr2)))