Распределите целое число по нескольким строкам столько раз, сколько оно разделено на константу

Question

У меня есть датафрейм

       Date      repair     
 <date>           <dbl>        
 2018-07-01        4420    
 2018-07-02          NA   
 2018-07-03          NA
 2018-07-04          NA
 2018-07-05          NA

Где 4420 - время в минутах. Я пытаюсь получить это:

       Date      repair     
 <date>           <dbl>        
 2018-07-01        1440    
 2018-07-02        1440   
 2018-07-03        1440
 2018-07-04         100
 2018-07-05          NA

Где 1440 - минут за один день и 100 что осталось. Я сделал это с петлей. Можно ли добиться этого более элегантным способом?

Answer 1

С dplyr:

library(dplyr)

df %>%
  mutate(
    repair = c(rep(1440, floor(repair[1] / 1440)), 
               repair[1] %% 1440, 
               rep(NA, n() - length(c(rep(1440, floor(repair[1] / 1440)), repair[1] %% 1440))))
  )

Выход:

        Date repair
1 2018-07-01   1440
2 2018-07-02   1440
3 2018-07-03   1440
4 2018-07-04    100
5 2018-07-05     NA

Answer 2

Вы можете написать небольшую функцию для этой задачи

f <- function(x, y, length_out) {
  remainder <- x %% y 
  if(remainder == 0) {
    `length<-`(rep(y, x %/% y), length_out)
  } else {
    `length<-`(c(rep(y, x %/% y), remainder), length_out)
  }
}

вход

x <- 4420
y <- 24 * 60

Результат

f(x, y, length_out = 10)
# [1] 1440 1440 1440  100   NA   NA   NA   NA   NA   NA

length_out должно быть равным nrow(your_data)

Answer 3

Рекурсивное решение:

fun <- function(x, y, i = 0){
  if(x <= y) c(rep(y, i), x) else fun(x-y, y, i+1)
}

fun(4420, 1440)[1:nrow(df)]
# [1] 1440 1440 1440  100   NA