我大致了解应用函数系列的功能,但是我在使用它来基于另一个缺少值的列来突变新列时遇到了麻烦。我可以使用for循环来完成任务,但是我想通过使用Apply类型函数来提高性能
说我有一个从今天开始到从现在开始几年结束的时间序列索引。我的原始索引仅存在于前几年。然后,我想在接下来的几年中使用假设的百分比变化(假设为10%)来人为地扩展这些索引,并将其存储为新列。
这是我的样本数据集:
data <- data.frame(
date = seq.Date(as.Date("2019-01-01"),as.Date("2021-01-01"),"3 months"),
index = c(1,1.2,1.4,1.5,1.6,1.7,NA,NA,NA)
)
我现在可以使用for循环创建新列index2:
data$index2 <- 1
for (i in 1:nrow(data)) {
if (!is.na(data$index[i])) {
data$index2[i] = data$index[i]
}
else {
data$index2[i] = data$index2[i-1]*1.1
}
}
但是,我不知道如何使用apply函数来完成此任务。再次感谢您的任何建议。
答案 0 :(得分:2)
假设我正确理解,这似乎是lag
的工作:
library(dplyr)
data %>% mutate(index2 = if_else(!is.na(index), index, lag(index) * 1.1))
# date index index2
#1 2019-01-01 1.0 1.00
#2 2019-04-01 1.2 1.20
#3 2019-07-01 1.4 1.40
#4 2019-10-01 1.5 1.50
#5 2020-01-01 1.6 1.60
#6 2020-04-01 1.7 1.70
#7 2020-07-01 NA 1.87
#8 2020-10-01 NA NA
#9 2021-01-01 NA NA
这将重现您的预期输出(即仅替换前一个NA
);我可能对您的问题陈述有误解,但我看不出*apply
与之有关。
您可以实现这样的sapply
位置
transform(data, index2 = c(index[1], sapply(seq_along(index)[-1], function(i)
if (!is.na(index[i])) index[i] else index[i - 1] * 1.1)))
# date index index2
#1 2019-01-01 1.0 1.00
#2 2019-04-01 1.2 1.20
#3 2019-07-01 1.4 1.40
#4 2019-10-01 1.5 1.50
#5 2020-01-01 1.6 1.60
#6 2020-04-01 1.7 1.70
#7 2020-07-01 NA 1.87
#8 2020-10-01 NA NA
#9 2021-01-01 NA NA
但这不是很漂亮。
修正拼写错误后,问题陈述会稍有变化,我们需要cumprod
data %>%
mutate(index2 = if_else(
!is.na(index),
index,
index[which.max(index)] * cumprod(c(rep(1.0, sum(!is.na(index))), rep(1.1, sum(is.na(index)))))))
# date index index2
#1 2019-01-01 1.0 1.0000
#2 2019-04-01 1.2 1.2000
#3 2019-07-01 1.4 1.4000
#4 2019-10-01 1.5 1.5000
#5 2020-01-01 1.6 1.6000
#6 2020-04-01 1.7 1.7000
#7 2020-07-01 NA 1.8700
#8 2020-10-01 NA 2.0570
#9 2021-01-01 NA 2.2627