Ускорение или замена для цикла: изменение формы группы и связывание строк - PullRequest
Новая
       50

Ускорение или замена для цикла: изменение формы группы и связывание строк

0 голосов
/ 22 сентября 2019

Добрый вечер, ребята, я хочу объединить одну и ту же строку идентификатора в одну строку с дополнительным столбцом, и вот моя часть данных. 

sample=structure(list(crsp_fundno = c(18021, 18021, 18021, 18021, 22436, 
                                      22436, 22436, 22436, 22436, 22436, 49805, 49805, 49805, 55603, 
                                      55603, 93362), seq = c(1L, 2L, 3L, 4L, 1L, 2L, 3L, 4L, 5L, 6L, 
                                                             1L, 2L, 3L, 1L, 2L, 1L), begdt = structure(c(13513, 14298, 15027, 
                                                                                                          16149, 12417, 13969, 14910, 14918, 15042, 15644, 14782, 14910, 
                                                                                                          15544, 15505, 15531, 17571), class = "Date"), enddt = structure(c(14297, 
                                                                                                                                                                            15026, 16148, 17621, 13968, 14909, 14917, 15041, 15643, 17621, 
                                                                                                                                                                            14909, 15543, 17621, 15530, 17621, 17621), class = "Date"), crsp_obj_cd = c("EDYG", 
                                                                                                                                                                                                                                                        "EDYG", "EDYG", "EDYG", "EDYG", "EDYG", "EDYG", "EDYG", "EDYG", 
                                                                                                                                                                                                                                                        "EDYG", "EF", "EF", "EF", "EDYB", "EDYB", "M"), lipper_class = c("MLGE", 
                                                                                                                                                                                                                                                                                                                         "MCCE", "MCVE", "MLCE", "MLVE", "MLVE", "MLCE", "MLVE", "MLCE", 
                                                                                                                                                                                                                                                                                                                         "MLVE", "IMLC", "IMLG", "IMLC", "MTAM", "MTAC", "MATJ"), lipper_obj_cd = c("G", 
                                                                                                                                                                                                                                                                                                                                                                                                    "G", "G", "G", "G", "G", "G", "G", "G", "G", "IF", "IF", "IF", 
                                                                                                                                                                                                                                                                                                                                                                                                    "GI", "GI", "I"), lipper_asset_cd = c("EQ", "EQ", "EQ", "EQ", 
                                                                                                                                                                                                                                                                                                                                                                                                                                          "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", "EQ", 
                                                                                                                                                                                                                                                                                                                                                                                                                                          "EQ")), class = "data.frame", row.names = c(NA, -16L))

Я попытался объединить строку с таким же идентификаторомв одном ряду, и вот мой код. 

temp=list()
dn=unique(sample$crsp_fundno)
for(i in 1:length(dn) ){
  part=sample[which(sample$crsp_fundno %in% dn[i]),]
  part=reshape(part,idvar='crsp_fundno',timevar='seq',direction='wide')
  temp[[i]]=part
}

library(plyr)
sum=rbind.fill(temp[[1]],temp[[2]])

for (i in 3 :length(dn)){sum=rbind.fill(sum,temp[[i]])}

Код работает, но слишком низкий во всех моих данных (94000 obs почти 2 часа).

Я думаю, что я не должен сильноположитесь на цикл for в большом наборе данных.

Может кто-нибудь знает, как я могу улучшить код или мою логику?

Спасибо за вашу помощь.

Ответы [ 2 ]

1 голос
/ 22 сентября 2019

Я настоятельно рекомендую data.table, когда речь заходит о скорости и эффективности памяти. 

setDT(sample) # in place, no assignment needed
sum3 <- dcast(sample,
              crsp_fundno ~ seq,
              value.var = names(sample)[3:8])

Вот сравнение OP 'for loop, способа reshape, предложенного в одномответов и способ data.table, предложенный в этом посте: 

Unit: milliseconds
       expr       min        lq      mean    median        uq       max neval cld
   for loop 23.735154 24.190626 25.948536 24.722330 26.176343 42.764253   100   c
    reshape  6.448800  6.742147  7.196820  6.850390  7.379401  9.932432   100  b 
 data.table  1.928812  2.143367  2.362979  2.255964  2.447935  5.847116   100 a
1 голос
/ 22 сентября 2019

Таким образом, использование reshape является правильным, однако реализация не идеальна.Функция уже оптимизирована для преобразования между форматами long и wide без необходимости использования цикла for.

Вам нужно всего лишь вызвать его один раз и сэкономить время: 

library(reshape2)
sum <- reshape(sample,direction = "wide",idvar = "crsp_fundno",timevar = "seq")

Как вы правильно поняли, reshape может плавно переключаться между форматами.

В вашем случае у вас есть:

  1. crsp_fundno - это переменная в формате long, которая идентифицирует несколько записей из одной группы
  2. seq - это переменная в формате long, которая различает несколько записей из одной группы
...