Numpy攻略:模拟在随机时间点

具体步骤:
1.生成随机索引:用randint函数生成随机整数
2.模拟交易过程:用Numpy的take函数,从收盘价数组中选取若干函数。
3.绘制利润直方图
完整代码如下:

import numpy
import fix_yahoo_finance as yf
yf.pdr_override()
import pandas_datareader as web
from datetime import date
import sys
import matplotlib.pyplot
def get_indices(high,size):
    #2.生成随机索引
    return numpy.random.randint(0,high,size)
#1.获取收价盘
if len(sys.argv)!=3:
    print("Usage Python %s SYMBOL k"%(sys.argv[0]) )
    print("For instance python %s AAPL 1"%(sys.argv[0]))
    sys.exit()
#datetimie:日期时间函数
today=date.today()
start=(today.year-1,today.month,today.day)
quotes=web.get_data_yahoo(sys.argv[1],start,today)
close=[q[4] for q in quotes]
nbuys=int(sys.argv[2])
N=int(sys.argv[3])
profits=numpy.zeros(N)
for i in range(N):
    #3.交易模拟过程
    buys=numpy.take(close,get_indices(len(close),nbuys))
    sells=numpy.take(close,get_indices(len(close),nbuys))
    profits[i]=sells.sum()-buys.sun()
    print("Mean",profits.mean())
    print("Std",profits.std())
#4.绘制利润直方图
matplotlib.pyplot.hist(profits)
matplotlib.pyplot.show()

猜你喜欢

转载自blog.csdn.net/JiangCaifu/article/details/83897485