1

假设我有一个名为 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))

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

4

2 回答 2

2

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"
于 2010-12-16T10:11:29.293 回答
0

如果您将打印语句更改为简单的字符向量本身，例如“Red Bar”，您可以将其添加到现有向量中，例如最后一行。如果您用 return() 代替 print()，代码可能会更清晰。唯一的问题是向量需要具有所有相同的模式，因此您需要接受字符向量或使用单行 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)
于 2010-12-15T22:55:22.157 回答