将R中的数据表中的连续行分组

时间:2018-12-19 18:56:59

标签: r data.table

我找不到这个简单问题的解决方案。 我有一个像这样的数据框:

df = data.table(X1 = 1:12, X2 = LETTERS[1:12])

我想创建一个名为group的新变量,在此我想将连续的行按3分组。例如:第1-3行将为group0,第4-6行将为group2,依此类推。

预期结果:

    X1 X2 group
 1:  1  A     0
 2:  2  B     0
 3:  3  C     0
 4:  4  D     1
 5:  5  E     1
 6:  6  F     1
 7:  7  G     2
 8:  8  H     2
 9:  9  I     2
10: 10  J     3
11: 11  K     3
12: 12  L     3

4 个答案:

答案 0 :(得分:6)

Mod运算符%/%是执行以下操作的简便方法:

df[, group := (1:.N - 1) %/% 3]
df
#    X1 X2 group
# 1:  1  A     0
# 2:  2  B     0
# 3:  3  C     0
# 4:  4  D     1
# 5:  5  E     1
# 6:  6  F     1
# 7:  7  G     2
# 8:  8  H     2
# 9:  9  I     2
#10: 10  J     3
#11: 11  K     3
#12: 12  L     3

答案 1 :(得分:5)

您也可以使用rep

df[, group := rep(1:(.N / 3) - 1, each = 3)]
df
#    X1 X2 group
# 1:  1  A     0
# 2:  2  B     0
# 3:  3  C     0
# 4:  4  D     1
# 5:  5  E     1
# 6:  6  F     1
# 7:  7  G     2
# 8:  8  H     2
# 9:  9  I     2
#10: 10  J     3
#11: 11  K     3
#12: 12  L     3

答案 2 :(得分:2)

这是一种方法。

consecutive_groups <- function(df, group_length = 3) {
  n_groups <- if_else(nrow(df) %% group_length == 0L, 
                      nrow(df) / group_length, 
                      as.integer(nrow(df) / group_length) + 1 )
  df %>% mutate(row_number = row_number(), 
                group = ntile(row_number, n_groups) - 1) %>% 
         select(-row_number)
}




 consecutive_groups(df, 3)
   #     X1 X2 group
   # 1   1  A     0
   # 2   2  B     0
   # 3   3  C     0
   # 4   4  D     1
   # 5   5  E     1
   # 6   6  F     1
   # 7   7  G     2
   # 8   8  H     2
   # 9   9  I     2
   # 10 10  J     3
   # 11 11  K     3
   # 12 12  L     3

答案 3 :(得分:0)

对于这种具有快速功能的情况:

   group <- function(dt,groupsize)
    {
      if(nrow(dt) %% groupsize != 0){
        print("Groupsize must be divisor of nrow!")
        return()
      }
    return(cbind(dt,data.frame(group=unlist(lapply(1:(nrow(dt) / groupsize),function(i){rep(i,groupsize)}))))))
    }