Вставка строк в фрейм данных на основе условий - PullRequest
Новая
       7

Вставка строк в фрейм данных на основе условий

0 голосов
/ 02 июня 2018

Мне нужно разделить начальную и конечную даты на интервалы в 30 дней и получить две новые переменные: «разделить дату начала» и «разделить дату окончания».

Я пробовал много вещей, но не сделалработа ... 

df_actual <- data.frame(
  order_id = "a", 
  start_date="01/05/2017", 
  ends_with="06/07/2017"
)


df_expected  <- data.frame(
  order_id = rep("a", 3), 
  start_date = rep("01/05/2017", 3), 
  ends_with = rep("06/07/2017", 3),
  split_start_date = c("01/05/2017", "31/05/2017", "30/06/2017"), 
  split_end_date = c("30/05/2017", "29/06/2017", "06/07/2017")
)

Ответы [ 2 ]

0 голосов
/ 02 июня 2018 
library(lubridate)

df_actual<-data.frame(order_id="a",start_date="01/05/2017",ends_with="06/07/2017")
start_date <- dmy(as.character(df_actual[1,2]))
ends_with <- dmy(as.character(df_actual[1,3]))

day_number <- as.numeric(difftime(ends_with, start_date))

batch_duration = 30

no_batches <- ceiling(day_number / batch_duration)
remainder <- day_number %% batch_duration + 1#for last day

batches <- rep(batch_duration, no_batches)
if(remainder > 0){
  batches[no_batches] <- remainder
}

new_cols <- data.frame(split_start_date=character(), split_end_date=character())
prev_end_date <- start_date - 1
for(i in 1:no_batches){
  split_end_date <- start_date + (i-1) * batch_duration + (batches[i]-1)
  split_start_date <- prev_end_date + 1
  prev_end_date <- split_end_date
  data_frame_row <- data.frame(split_start_date, split_end_date)
  new_cols <- rbind(new_cols, data_frame_row)
}

# print(new_cols)
print(cbind(df_actual, new_cols))
0 голосов
/ 02 июня 2018

Я увеличил ваш пример, добавив 2 order_ids, чтобы посмотреть, что произойдет, если будет более 1 идентификатора.

Я создал функцию my_fun, смотрите раздел function.В этой функции я беру start_date и создаю новые start_dates на основе разницы между start_date и end_withs, деленной на продолжительность.Это дает новые даты, которые должны быть созданы.после того, как даты split_start были созданы, даты split_end - просто упражнение заполнения, стараясь вычесть 1 день, чтобы добраться до выходных данных вашего примера.

Использование map2 с setNames и bind_rows создаетdata.frame с новыми датами и order_id в качестве идентификатора.затем его можно объединить с исходной таблицей для создания ожидаемого результата.

Редактировать:

добавлен код для работы с граничными случаями, такими как NA в датах или когда start_date равен end_with. 

library(lubridate)
library(dplyr)
library(purrr)

df_actual %>% 
  inner_join(map2(df_actual$start_date, df_actual$ends_with, my_fun) %>% 
               setNames(., df_actual$order_id) %>% 
               bind_rows(., .id = "order_id"))

  order_id start_date  ends_with split_start  split_end
1        a 2017-05-01       <NA>        <NA>       <NA>
2        b 2016-05-01 2016-07-06  2016-05-01 2016-05-30
3        b 2016-05-01 2016-07-06  2016-05-31 2016-06-29
4        b 2016-05-01 2016-07-06  2016-06-30 2016-07-06
5        c 2017-07-01 2017-07-01  2017-07-01 2017-07-01
6        d       <NA> 2017-07-01        <NA>       <NA>

функция: 

my_fun <- function(x, y, duration = 30) {
  d <- ddays(duration)

  if(is.na(x) | is.na(y)) split_start <- split_end <- NA else 
    if(x == y) {
      split_start <- x
      split_end <- y
      } else {
        n <- (y - x) %/% d 

        split_start <- c(rep(x, n + 1))

        for(i in 1:n+1){
          split_start[i] <- x + (i - 1) * d
        }

        split_end <- split_start[1:n+1]
        split_end <- split_end - ddays(1)
        split_end[n+1] <- y
        }

  df <- data.frame(split_start, split_end)
  return(df)
}

данные: 

df_actual <- structure(list(order_id = c("a", "b", "c", "d"), start_date = structure(c(17287, 
16922, 17348, NA), class = "Date"), ends_with = structure(c(NA, 
16988, 17348, 17348), class = "Date")), class = "data.frame", row.names = c(NA, 
-4L))
...