基于另一列更改ID值

Question

我有一个很大的数据集，看起来像这样：

Conv.  Rev.  ID  Order  path_no
 0     0     1    1        1
 1     50    1    2        1
 0     0     1    3        2
 1     100   1    4        2
 0     0     2    1        1
 0     0     2    2        1
 1     150   2    3        1
 1     100   2    4        2

当有新的path_no时，我想创建一个新的ID列，然后ID将会改变。所以我希望它看起来像这样：

Conv.  Rev.  ID  Order  path_no
  0     0     1    1        1
  1     50    1    2        1
  0     0     2    3        2
  1     100   2    4        2
  0     0     3    1        1
  0     0     3    2        1
  1     150   3    3        1
  1     100   4    4        2

Answer 1

我认为data.table的rleid应该能做到这一点。这里有一个使用data.table和dplyr的解决方案

dplyr::mutate(df, ID = data.table::rleid(path_no))

  Conv. Rev. ID Order path_no
1     0    0  1     1       1
2     1   50  1     2       1
3     0    0  2     3       2
4     1  100  2     4       2
5     0    0  3     1       1
6     0    0  3     2       1
7     1  150  3     3       1
8     1  100  4     4       2

或者仅使用data.table：

dt <- setDT(df)
dt[, ID := rleid(path_no)][]

   Conv. Rev. ID Order path_no
1:     0    0  1     1       1
2:     1   50  1     2       1
3:     0    0  2     3       2
4:     1  100  2     4       2
5:     0    0  3     1       1
6:     0    0  3     2       1
7:     1  150  3     3       1
8:     1  100  4     4       2

数据：

text <- "Conv.  Rev.  ID  Order  path_no
        0     0     1    1        1
        1     50    1    2        1
        0     0     1    3        2
        1     100   1    4        2
        0     0     2    1        1
        0     0     2    2        1
        1     150   2    3        1
        1     100   2    4        2"

df <- read.table(text = text, stringsAsFactors = FALSE, header = TRUE)

Answer 2

可以使用一个简单的for循环：

vals <- c(1, 1, 1, 2, 2, 2, 1, 1, 2)
nobs <- length(vals)
idx <- rep(1, nobs)
for (i in 2:nobs) {
   if (vals[i] != vals[i-1]) {
       idx[i] <- idx[i-1] + 1
   } else {
       idx[i] <- idx[i-1]
   }
 }