Python编程掘金3量化交易之各技术指标的实现

coding=utf-8

from future import print_function, absolute_import, unicode_literals
import numpy as np
import pandas as pd
import tushare as ts
#import matplotlib.pyplot as plt
import talib
from gm.api import *
from datetime import timedelta

def init(context):
today = context.now
# context.goods交易的品种 上证：SHSE.600519 深证：SZSE.000651 SHFE.rb2010
print (‘initinitinitinitinitinitinitinitinitinitinitinitinitinitinitinitinitinitinitinit’)
schedule(schedule_func=algo1, date_rule=‘1d’, time_rule=‘09:15:00’)
schedule(schedule_func=algo2, date_rule=‘1d’, time_rule=‘09:30:00’)
#context.goods =[‘SZSE.000002’,‘SZSE.000049’,‘SZSE.000661’,‘SZSE.000828’,‘SZSE.000848’,‘SZSE.000858’,‘SZSE.000895’,‘SZSE.000921’,‘SZSE.002007’,‘SZSE.002117’,‘SZSE.002127’,‘SZSE.002146’,‘SZSE.002179’,‘SZSE.002236’,‘SZSE.002242’,‘SZSE.002262’,‘SZSE.002372’,‘SZSE.002555’,‘SZSE.002572’,‘SZSE.002595’,‘SZSE.002677’,‘SZSE.002690’,‘SZSE.300033’,‘SZSE.300136’,‘SZSE.300357’,‘SZSE.300406’,‘SHSE.600036’,‘SHSE.600048’,‘SHSE.600273’,‘SHSE.600276’,‘SHSE.600452’,‘SHSE.600511’,‘SHSE.600519’,‘SHSE.600563’,‘SHSE.600566’,‘SHSE.600867’,‘SHSE.600987’,‘SHSE.601318’,‘SHSE.603288’,‘SHSE.603899’]

#context.goods = ['SZSE.159915']


#context.goods =['SZSE.000002','SZSE.000049','SZSE.000661','SZSE.000828','SZSE.000848','SZSE.000858','SZSE.000895','SZSE.000921','SZSE.002007','SZSE.002117','SZSE.002127','SZSE.002146','SZSE.002179','SZSE.002236','SZSE.002242','SZSE.002262','SZSE.002372','SZSE.002555','SZSE.002572','SZSE.002595','SZSE.002677','SZSE.002690','SZSE.300033','SZSE.300136','SZSE.300357','SZSE.300406','SHSE.600036','SHSE.600048','SHSE.600273','SHSE.600276','SHSE.600452','SHSE.600511','SHSE.600519','SHSE.600563','SHSE.600566','SHSE.600867','SHSE.600987','SHSE.601318','SHSE.603288','SHSE.603899']

'''
context.goods =['SHSE.600007','SHSE.600062','SHSE.600167','SHSE.600273','SHSE.600329','SHSE.600338',
                'SHSE.600352','SHSE.600373','SHSE.600566','SHSE.600598','SHSE.600639','SHSE.600668',
                'SHSE.600750','SHSE.600993','SHSE.601811','SHSE.601828','SHSE.603156','SHSE.603198',
                'SHSE.603357','SHSE.603360','SHSE.603367','SHSE.603568','SHSE.603587','SHSE.603599',
                'SHSE.603610','SHSE.603639','SHSE.603693','SHSE.603816','SHSE.603858','SZSE.000011',
                'SZSE.000090','SZSE.000544','SZSE.000582','SZSE.000623','SZSE.000672','SZSE.000789',
                'SZSE.000869','SZSE.000935','SZSE.000963','SZSE.000999','SZSE.002016','SZSE.002019',
                'SZSE.002020','SZSE.002039','SZSE.002128','SZSE.002191','SZSE.002223','SZSE.002233',
                'SZSE.002234','SZSE.002275','SZSE.002353','SZSE.002399','SZSE.002444','SZSE.002458',
                'SZSE.002511','SZSE.002601','SZSE.002605','SZSE.002737','SZSE.002803','SZSE.002818',
                'SZSE.002869','SZSE.300031','SZSE.300360','SZSE.300401','SZSE.300418','SZSE.300427',
                'SZSE.300462','SZSE.300481','SZSE.300552','SZSE.300571','SZSE.300607','SZSE.300702',
                'SZSE.300758','SZSE.300771','SZSE.300776']
'''

def algo1(context):
print (‘algo1algo1++++++++++++++++algo1algo1+++++++++++++++++++++++++++++algo1algo1++++++++++++++++++++++++++++++++algo1algo1+’)
print (‘algo1algo1++++++++++++++++algo1algo1+++++++++++++++++++++++++++++algo1algo1++++++++++++++++++++++++++++++++algo1algo1+’)
# 获取当前时间
today = context.now

# 获取上一个交易日
last_day = get_previous_trading_date(exchange='SHSE', date=today)
#symbol='context.goods'
# 选取全A股（剔除停牌和st股和上市不足50日的新股和退市股和B股）
date1 = (context.now - timedelta(days=100)).strftime("%Y-%m-%d %H:%M:%S")
date2 = context.now.strftime("%Y-%m-%d %H:%M:%S")
all_stock = get_instruments(exchanges='SHSE, SZSE', sec_types=[1], fields='symbol, listed_date, delisted_date',
                            df=True)
print ('today ',today,'当天所有股票数 is ', len(all_stock['symbol']) )                                
context.code = all_stock[(all_stock['listed_date'] < date1) & (all_stock['delisted_date'] > date2) &
                        (all_stock['s