Есть колонка в датафрейме примерно с таким содержанием (на самом деле количество категорий идущих подряд не равно)
data <- data.frame(Activity = rep(rep(x = c("lookout", "action", "sleep"), each = 10), 5))
head(data, 8)
Activity
1 lookout
2 lookout
3 lookout
4 lookout
5 lookout
6 lookout
7 action
8 actionЯ бы хотел напротив каждой категории написать какая она по счёту в блоке без смены на другую категорию, то есть все первые 6 lookout будет с единицей и action тоже, а когда дойдёт до новых lookout и action, то чтобы была цифра должна быть 2 и так далее:
Activity order
1 lookout 1
2 lookout 1
3 lookout 1
4 lookout 1
5 lookout 1
6 lookout 1
7 action 1
8 action 1
9 action 1
10 action 1
11 action 1
12 action 1
13 sleep 1
14 sleep 1
15 sleep 1
16 sleep 1
17 sleep 1
18 sleep 1
19 lookout 2
20 lookout 2
21 lookout 2
22 lookout 2
23 lookout 2
24 lookout 2
25 action 2
26 action 2
27 action 2
28 action 2
29 action 2
30 action 2
Я написал функцию с for, но мне она не нравится. Можно ли как-то иначе сделать?
set.numerator <- function(x) {
result <- vector(mode = "integer", length = length(x))
stage_numbers <- vector(mode = "integer", length = n_distinct(x))
names(stage_numbers) <- unique(x)
current.stage <- "undefined"
n_stage <- 0
for (idx in seq(x)) {
row.stage <- x[idx]
if(current.stage != row.stage) {
current.stage <- row.stage
stage_numbers[row.stage] = stage_numbers[row.stage] + 1
n_stage <- stage_numbers[row.stage]
}
result[idx] <- n_stage
}
return(result)
}
