Question

假设我有一个名为TLT的data.frame，其最后一行是：

           TLT.Open  TLT.Close 
2010-12-14     92.4      92.14

我想添加一个名为TLT.BarColor的额外矢量，所以它看起来像这样：

           TLT.Open  TLT.Close  TLT.BarColor
2010-12-14     92.4      92.14       "Green"

这是一个“打印”的功能，无论是绿色还是红色的日期。

bar_color <- function(ticker) {

require("quantmod")

x <- getSymbols(ticker, auto.assign=FALSE)

open        <- x[,1]                       
close       <- x[,2]                       

last_open   <- tail(open,  n=1)            
last_close  <- tail(close, n=1)            



if       (last_open > last_close)    
           {print("Red Bar")} 

else if  (last_open < last_close)          
           {print("Green Bar")}   

 else       {print("Doji Bar")}

不是使用print（）R函数（只打印到控制台），而是使用什么R函数发送输出以填充新的向量？

super_dataframe <- cbind(TLT, apply(TLT, 1, valid_function))

示例函数在此解决方案中不起作用。但如果函数有效，它的输出可以这种方式附加。

Answer 1

ticker不能是数据框，但必须是一个字符。因此，使用您创建超级数据框的应用程序，您将遇到问题。以下功能为不同的代码提供标签。

bar_color <- function(ticker){
   x <- getSymbols(ticker,auto.assign=FALSE)
   n <- nrow(x)
   switch(
      sign(x[n,1]-x[n,4])+2,
      "Green Bar",
      "Doji Bar",
      "Red Bar")
}

> TLT <- c("F","QQQQ")
> cbind(TLT,sapply(TLT,bar_color))
     TLT               
F    "F"    "Green Bar"
QQQQ "QQQQ" "Red Bar"

如果您想要一个自动收报机但不同日期的标签，那么这就是您所需要的：

bar_color <- function(ticker){
   x <- as.data.frame(getSymbols(ticker,auto.assign=FALSE))

   x$barcolor <- sapply(
            as.numeric(sign(x[,1]-x[,4])+2),
            function(j) switch(j,"Green Bar","Doji Bar","Red Bar")
   )

   return(x)
}

> head(bar_color("F"))
           F.Open F.High F.Low F.Close F.Volume F.Adjusted  barcolor
2007-01-03   7.56   7.67  7.44    7.51 78652200       7.51   Red Bar
2007-01-04   7.56   7.72  7.43    7.70 63454900       7.70 Green Bar
2007-01-05   7.72   7.75  7.57    7.62 40562100       7.62   Red Bar
2007-01-08   7.63   7.75  7.62    7.73 48938500       7.73 Green Bar
2007-01-09   7.75   7.86  7.73    7.79 56732200       7.79 Green Bar
2007-01-10   7.79   7.79  7.67    7.73 42397100       7.73   Red Bar

你可能面临的问题是，getSymbols不会返回数据帧，而是返回xts对象。对于xts，有一些访问和添加数据的特定方法，人们不应期望它的行为类似于数据框。

> X <- getSymbols("F",auto.assign=FALSE)
> class(X)
[1] "xts" "zoo"

Answer 2

如果您将print语句更改为字符向量本身，例如“Red Bar”，则可以将其添加到现有向量（例如最后一行）。如果你用print（）替换return（），它可能是更清晰的代码。唯一的问题是向量需要具有相同的模式，因此您需要接受一个字符向量或使用一行data.frame。

vec <- c(TLT[NROW(TLT), ] , bar.color( "TLT") )  # a character vector

onerowdf <- cbind( TLT[NROW(TLT), ], bar.color( "TLT")) ) 
# a data.frame (aka list)

ARE函数将输出发送到新向量？

2 个答案: