如何優雅地用不同的矢量替代重複值？

Question

在R，這樣的載體：

x <- c(0,0,1,1,0,0,1,1,0,0,1,1,1,1,0,1,0,1,0,1) 

我想X是這樣的：

y = c(0,0,1,1,0,0,2,2,0,0,3,3,3,3,0,4,0,5,0,6) 

誰能幫我解決這個問題嗎？

Answer 1

的選擇是rle

inverse.rle(within.list(rle(x), values[values!=0] <- seq_along(values[values!=0]))) 
#[1] 0 0 1 1 0 0 2 2 0 0 3 3 3 3 0 4 0 5 0 6 

---

或者與rleid從data.table

library(data.table) 
x1 <- x*rleid(x) 
match(x1, unique(x1[x1!=0]), nomatch = 0) 
#[1] 0 0 1 1 0 0 2 2 0 0 3 3 3 3 0 4 0 5 0 6