R - 更优雅的代码行编写方式

Question

好的，让我们使用下面的代码计算 2 天内的滚动简单移动平均线：

# Use TTR package to create rolling SMA n day moving average 
new.df$close.sma.n2 <- SMA(new.df[,"Close"], 2)

假设我要计算 2:30 的 n 天时间

这里的输入是：

close.sma.n** 以及 SMA 计算的数值。

所以我的问题是： 如何编写一行代码以在不同的 SMA 周期上执行上述计算，并在数据框中创建一个具有相应 close.sma.n2、3、4、5、6 等值的新列？

我知道我能做到：

n.sma <- 2:30

并将该变量放入：

new.df$close.sma.n2 <- SMA(new.df[,"Close"], n.sma)

也许我可以：

name <- "n2:30"

并把它放在里面：

 new.df$close.sma.name <- SMA(new.df[,"Close"], n.sma)

Answer 1

您没有提供示例数据或 SMA，因此我编写了虚拟函数来测试我的代码。

df <- data.frame(Close=c(1, 2, 3, 4))
SMA <- function(x, numdays) {numdays}

然后，我编写了一个函数，该函数需要平均天数，并返回一个函数，该函数需要一个 data.frame 并在那么多天内获取 SMA。

getSMA <- function(numdays) {
  function(new.df) {
    SMA(new.df[,"Close"], numdays)
  }
}

然后，创建一个矩阵来放入 SMA

smas <- matrix(nrow=nrow(df), ncol=0)

然后填写。

for (i in 2:30) {
  smas <- cbind(smas, getSMA(i)(df))
}

然后，将列名设置为您想要的名称

colnames(smas) <- sapply(2:30, function(n)paste("close.sma.n", n, sep=""))

并将其与起始数据框绑定。

df <-  cbind(df, smas)

现在您有了原始的 data.frame，加上 "close.sma.n2" through ".n30"