talib_testcase.py 文件源码

def baseAPI():
    #??tushare??????
    df=ts.get_k_data('300580',start='2017-01-12',end='2017-05-26')
    #?????
    closed=df['close'].values
    #??????????timeperiod??????? 5,10,20 ???????
    ma5=talib.SMA(closed,timeperiod=5)
    ma10=talib.SMA(closed,timeperiod=10)
    ma20=talib.SMA(closed,timeperiod=20)

    #?????????
    print closed
    print ma5
    print ma10
    print ma20

    #??plog????????????????
    plt.plot(closed)
    plt.plot(ma5)
    plt.plot(ma10)
    plt.plot(ma20)
    #??????????????????
    plt.grid()
    #???????????????
    plt.show()