参加了此次wot分享,对R语言预测股票走势很感兴趣,回来后做了点功课,对例子中的没有提供源码的部分做了补充,分享给大家:
# 散点数据
genPoint<-function(pdata,ldata){pdata<-data.frame(Index=as.Date(array(index(ldata))),Series=array(ifelse(ldata$ma20>ldata$Value,"up","down")),Value=array(ldata$ma20))
pdata
}
#交易信号
Signal<-function(cdata,pdata){
p=0
r0<-NULL
r1<-NULL
rn<-NULL
for(i in 1:nrow(pdata)) {
if(pdata$Series[i]=='down') {
if( p==0) {
r0<-c(r0,"B")
r1<-c(r1,cdata$Close[i])
rn<-c(rn,index(cdata[i]))
}
p<-1
}
if(pdata$Series[i]=='up'){
if( p==1) {
r0<-c(r0,"S")
r1<-c(r1,cdata$Close[i])
rn<-c(rn,index(cdata[i]))
}
p<-0
}
}
res<-data.frame(Value=r1,op=r0)
rownames(res)<-as.Date(rn)
return(res)
}
#交易信号
Signal<-function(cdata,pdata){
p=0
r0<-NULL
r1<-NULL
rn<-NULL
for(i in 1:nrow(pdata)) {
if(pdata$Series[i]=='down') {
if( p==0) {
r0<-c(r0,"B")
r1<-c(r1,pdata$Value[i])
rn<-c(rn,as.Date(index(cdata[i])))
}
p<-1
}
if(pdata$Series[i]=='up'){
if( p==1) {
r0<-c(r0,"S")
r1<-c(r1,pdata$Value[i])
rn<-c(rn,as.Date(index(cdata[i])))
}
p<-0
}
}
res<-data.frame(Value=r1,op=r0)
rownames(res)<-as.Date(rn)
}