Нумерация строк в группах в кадре данных

163

Работа с фреймом данных, похожим на этот:

set.seed(100)  
df <- data.frame(cat = c(rep("aaa", 5), rep("bbb", 5), rep("ccc", 5)), val = runif(15))             
df <- df[order(df$cat, df$val), ]  
df  

   cat        val  
1  aaa 0.05638315  
2  aaa 0.25767250  
3  aaa 0.30776611  
4  aaa 0.46854928  
5  aaa 0.55232243  
6  bbb 0.17026205  
7  bbb 0.37032054  
8  bbb 0.48377074  
9  bbb 0.54655860  
10 bbb 0.81240262  
11 ccc 0.28035384  
12 ccc 0.39848790  
13 ccc 0.62499648  
14 ccc 0.76255108  
15 ccc 0.88216552 

Я пытаюсь добавить столбец с нумерацией в каждой группе. Делая это таким образом, очевидно, не используя полномочия R:

 df$num <- 1  
 for (i in 2:(length(df[,1]))) {  
   if (df[i,"cat"]==df[(i-1),"cat"]) {  
     df[i,"num"]<-df[i-1,"num"]+1  
     }  
 }  
 df  

   cat        val num  
1  aaa 0.05638315   1  
2  aaa 0.25767250   2  
3  aaa 0.30776611   3  
4  aaa 0.46854928   4  
5  aaa 0.55232243   5  
6  bbb 0.17026205   1  
7  bbb 0.37032054   2  
8  bbb 0.48377074   3  
9  bbb 0.54655860   4  
10 bbb 0.81240262   5  
11 ccc 0.28035384   1  
12 ccc 0.39848790   2  
13 ccc 0.62499648   3  
14 ccc 0.76255108   4  
15 ccc 0.88216552   5  

Что было бы хорошим способом сделать это?

эли-к
источник
1
Я бы предложил добавить что-то вроде «seq вдоль уровней» или «считать по повторениям» в заголовке вопроса, так как я нашел этот вопрос, и это именно то, что я искал
crazysantaclaus
2
@crazysantaclaus Если бы это было название, я бы не нашел то, что искал :-( Я буквально искал "как
нумеровать

Ответы:

280

Используйте ave, ddply, dplyrили data.table:

df$num <- ave(df$val, df$cat, FUN = seq_along)

или:

library(plyr)
ddply(df, .(cat), mutate, id = seq_along(val))

или:

library(dplyr)
df %>% group_by(cat) %>% mutate(id = row_number())

или (самый эффективный в использовании памяти, поскольку он назначается ссылкой внутри DT):

library(data.table)
DT <- data.table(df)

DT[, id := seq_len(.N), by = cat]
DT[, id := rowid(cat)]
mnel
источник
2
Возможно, стоит упомянуть, что aveздесь вместо числа int выдается float. Поочередно, может измениться df$valна seq_len(nrow(df)). Я только что столкнулся с этим здесь: stackoverflow.com/questions/42796857/…
Фрэнк
1
Интересно, что это data.tableрешение кажется более быстрым, чем использование frank: library(microbenchmark); microbenchmark(a = DT[, .(val ,num = frank(val)), by = list(cat)] ,b =DT[, .(val , id = seq_len(.N)), by = list(cat)] , times = 1000L)
hannes101
4
Спасибо! dplyrРешение хорошо. Но если, как и я, вы продолжали получать странные ошибки при попытке такого подхода, убедитесь, что вы не получаете конфликтов между ними plyrи dplyrкак описано в этом посте. Этого можно избежать, явно вызвавdplyr::mutate(...)
EcologyTom
2
другой data.tableметодsetDT(df)[, id:=rleid(val), by=.(cat)]
chinsoon12
Как изменить library(plyr)и library(dplyr)ответить, чтобы сделать столбец рейтинга val в порядке убывания?
Пшемыслав Ремин
26

Для этого вопрос более полный, базовый вариант R с sequenceи rle:

df$num <- sequence(rle(df$cat)$lengths)

который дает намеченный результат:

> df
   cat        val num
4  aaa 0.05638315   1
2  aaa 0.25767250   2
1  aaa 0.30776611   3
5  aaa 0.46854928   4
3  aaa 0.55232243   5
10 bbb 0.17026205   1
8  bbb 0.37032054   2
6  bbb 0.48377074   3
9  bbb 0.54655860   4
7  bbb 0.81240262   5
13 ccc 0.28035384   1
14 ccc 0.39848790   2
11 ccc 0.62499648   3
15 ccc 0.76255108   4
12 ccc 0.88216552   5

Если df$catэто фактор-переменная, вам нужно as.characterсначала обернуть ее :

df$num <- sequence(rle(as.character(df$cat))$lengths)
Яап
источник
Только заметил, это решение требует catсортировки столбца?
zx8754
@ zx8754 да, если вы не хотите пронумеровать последовательными вхождениямиcat
Яап
9

Вот вариант использования forцикла по группам, а не по строкам (как OP)

for (i in unique(df$cat)) df$num[df$cat == i] <- seq_len(sum(df$cat == i))
маленький мальчик
источник
9

Вот небольшой трюк для улучшения, который позволяет сортировать 'val' внутри групп:

# 1. Data set
set.seed(100)
df <- data.frame(
  cat = c(rep("aaa", 5), rep("ccc", 5), rep("bbb", 5)), 
  val = runif(15))             

# 2. 'dplyr' approach
df %>% 
  arrange(cat, val) %>% 
  group_by(cat) %>% 
  mutate(id = row_number())
Андрий
источник
Вы не можете сортировать после group_by?
zcoleman
6

Я хотел бы добавить data.tableвариант, используя rank()функцию, которая предоставляет дополнительную возможность изменить порядок и, таким образом, делает его немного более гибким, чем seq_len()решение, и очень похож на функции row_number в RDBMS.

# Variant with ascending ordering
library(data.table)
dt <- data.table(df)
dt[, .( val
   , num = rank(val))
    , by = list(cat)][order(cat, num),]

    cat        val num
 1: aaa 0.05638315   1
 2: aaa 0.25767250   2
 3: aaa 0.30776611   3
 4: aaa 0.46854928   4
 5: aaa 0.55232243   5
 6: bbb 0.17026205   1
 7: bbb 0.37032054   2
 8: bbb 0.48377074   3
 9: bbb 0.54655860   4
10: bbb 0.81240262   5
11: ccc 0.28035384   1
12: ccc 0.39848790   2
13: ccc 0.62499648   3
14: ccc 0.76255108   4

# Variant with descending ordering
dt[, .( val
   , num = rank(-val))
    , by = list(cat)][order(cat, num),]
hannes101
источник
5

Другая dplyrвозможность может быть:

df %>%
 group_by(cat) %>%
 mutate(num = 1:n())

   cat      val   num
   <fct>  <dbl> <int>
 1 aaa   0.0564     1
 2 aaa   0.258      2
 3 aaa   0.308      3
 4 aaa   0.469      4
 5 aaa   0.552      5
 6 bbb   0.170      1
 7 bbb   0.370      2
 8 bbb   0.484      3
 9 bbb   0.547      4
10 bbb   0.812      5
11 ccc   0.280      1
12 ccc   0.398      2
13 ccc   0.625      3
14 ccc   0.763      4
15 ccc   0.882      5
tmfmnk
источник
3
В некоторых случаях вместо 1:n()использования seq_len(n())безопаснее, в случае, если в вашей последовательности операций у вас есть ситуация, в которой вы n()можете вернуться 0, потому что 1:0вы получаете вектор длины два, а вектор seq_len(0)нулевой длины, что позволяет избежать ошибки несоответствия длины с mutate().
Брайан Стэмпер
0

Использование rowid()функции в data.table:

> set.seed(100)  
> df <- data.frame(cat = c(rep("aaa", 5), rep("bbb", 5), rep("ccc", 5)), val = runif(15))
> df <- df[order(df$cat, df$val), ]  
> df$num <- data.table::rowid(df$cat)
> df
   cat        val num
4  aaa 0.05638315   1
2  aaa 0.25767250   2
1  aaa 0.30776611   3
5  aaa 0.46854928   4
3  aaa 0.55232243   5
10 bbb 0.17026205   1
8  bbb 0.37032054   2
6  bbb 0.48377074   3
9  bbb 0.54655860   4
7  bbb 0.81240262   5
13 ccc 0.28035384   1
14 ccc 0.39848790   2
11 ccc 0.62499648   3
15 ccc 0.76255108   4
12 ccc 0.88216552   5
AKRosenblad
источник
1
Спасибо за ваш ответ, но, похоже, он уже описан в последнем предложении в ответе @ mnel
eli-k