Определить значение по предыдущей и фактической строке - PullRequest
Новая
       64

Определить значение по предыдущей и фактической строке

0 голосов
/ 22 января 2019

У меня есть data.table с двумя полями, startvalue и endValue, которые мне нужно заполнить на основе некоторой информации из предыдущих и фактических строк.Хотя это как-то похоже на это и это , я не смог получить желаемые результаты.

Фиктивные данные: 

a <- data.table(user = c("A", "A", "A", "B", "B"), 
                gap = c(1, 0, 2, 2, 3), 
                priority = c(1, 3, 2, 2, 1))

Затем я фиксирую startValue для всех приоритетов == 1: 

setkey(a, user, priority)
a[priority == 1, startValue := 0]

, и я устанавливаю endValue для тех, которые startValue уже определены: 

a[!is.na(startValue), endValue := startValue + gap*3]

Теперь приходит проблема.Я хочу, чтобы startValue в строке 2 (пользователь A, приоритет 2) был таким же, как endValue в строке 1, чтобы я мог вычислить новый endValue.Я знаю Я могу использовать цикл, но я хотел знать, возможно ли это сделать с помощью любой другой функции или комбинации функций.

Я пробовал несколько комбинаций shift иzoo:na.locf но всегда заканчивалось путаницей уже существующих значений.

Ожидаемый результат: 

b <- structure(list(user = c("A", "A", "A", "B", "B"), 
                    gap = c(1, 2, 0, 3, 2), 
                    priority = c(1, 2, 3, 1, 2), 
                    startValue = c(0, 3, 9, 0, 9), 
                    endValue = c(3, 9, 9, 9, 15)), 
               row.names = c(NA, -5L), 
               class = c("data.table", "data.frame"))

Ответы [ 2 ]

0 голосов
/ 22 января 2019

Сначала вычислите конечные значения, используя cumsum.Затем используйте shift, чтобы получить начальные значения.

a[ , c("startValue", "endValue") := {
  e1 <- startValue[1] + gap[1] * 3
  endValue <- c(e1, e1 + cumsum(gap[-1] * 3))
  startValue <- shift(endValue, fill = startValue[1])
  .(startValue, endValue)
}, by = user]

#    user gap priority startValue endValue
# 1:    A   1        1          0        3
# 2:    A   2        2          3        9
# 3:    A   0        3          9        9
# 4:    B   3        1          0        9
# 5:    B   2        2          9       15
0 голосов
/ 22 января 2019

Мы можем использовать accumulate из purrr

library(purrr)
library(data.table)
a[, endValue := accumulate(gap,  ~   .x + .y * 3, .init = 0)[-1], user
   ][, startValue := shift(endValue, fill = 0), user][]
all.equal(a, b, check.attributes = FALSE)
#[1] TRUE

Или использовать Reduce из base R, чтобы создать столбец 'endValue', а затем взять lag из'endValue' для создания 'startValue', сгруппированного по 'user' 

a[, endValue := Reduce(function(x, y) x + y *3, gap, 
     accumulate = TRUE, init = 0)[-1], user]
...