Lag (Hmisc) с аргументом сдвига, определенным в другом столбце. R

Question

Чтобы сделать его простым, я хотел бы, чтобы столбец C был равен . Задержка столбца B с аргументом сдвига в зависимости от целых чисел в столбец A, так что я хочу:

A | B | C 
    0 | 5 | 5 
    2 | 6 | NA 
    3 | 7 | NA 
    2 | 8 | 6 

Я пробовал:

library(dplyr) 
library(Hmisc) 
data <- mutate(data, 
    C= Lag(B, shift=as.integer(A)), 

, но он не работает, я получаю NAs только, это, наверное, вопрос типа, но я не уверен, так как даже с as.integer он не работает, сом у кого-нибудь есть идея, почему это не работает?

Большое спасибо

Answer 1

lag из dplyr также имеет сдвиг аргумента, поэтому нет необходимости загружать другой пакет. Один из способов сделать это было бы,

library(dplyr) 
df$C <- diag(sapply(df$A, function(i) lag(df$B, i))) 
df 
# A B C 
#1 0 5 5 
#2 2 6 NA 
#3 3 7 NA 
#4 2 8 6 

Answer 2

Решение: Сдвиг аргумент не кажется, взять значения столбцов в виде целых чисел, , так что я сделал это по-другому:

df$C <- apply(as.data.frame(df$A), 1, function(i){ 
    nth(df$B, i) 
}) 

Он принимает п-я величина в, с п, находясь в седловине

(Большое спасибо за помощь мне SOTOS)