我找不到这个简单问题的解决方案。 我有一个像这样的数据框:
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
答案 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)}))))))
}