-
Notifications
You must be signed in to change notification settings - Fork 302
/
Copy pathKDJ指标策略.py
120 lines (103 loc) · 5.88 KB
/
KDJ指标策略.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
# 注：该策略仅供参考和学习，不保证收益。
# !/usr/bin/env python
# -*- coding: utf-8 -*-
# 策略代码总共分为三大部分，1)PARAMS变量 2)initialize函数 3)handle_data函数
# 请根据指示阅读。或者直接点击运行回测按钮，进行测试，查看策略效果。
# 策略名称：KDJ指标策略
# 策略详细介绍：https://wequant.io/study/strategy.kdj.html
# 关键词：随机指标、中短期。
# 方法：
# 1)通过特定周期内的最高价、最低价、收盘价以及移动平均线法计算KDJ值；
# 2)用KDJ判断超买超卖线，超买则卖出，超卖则买入。
import numpy as np
import talib
# 阅读1，首次阅读可跳过:
# PARAMS用于设定程序参数，回测的起始时间、结束时间、滑点误差、初始资金和持仓。
# 可以仿照格式修改，基本都能运行。如果想了解详情请参考新手学堂的API文档。
PARAMS = {
"start_time": "2016-01-01 00:00:00", # 回测起始时间
"end_time": "2016-05-01 00:00:00", # 回测结束时间
"commission": 0.002, # 此处设置交易佣金
"slippage": 0.001, # 此处设置交易滑点
"account_initial": {"huobi_cny_cash": 100000,
"huobi_cny_btc": 0}, # 设置账户初始状态
}
# 阅读2，遇到不明白的变量可以跳过，需要的时候回来查阅:
# initialize函数是两大核心函数之一（另一个是handle_data），用于初始化策略变量。
# 策略变量包含：必填变量，以及非必填（用户自己方便使用）的变量
def initialize(context):
# 设置回测频率, 可选："1m", "5m", "15m", "30m", "60m", "4h", "1d", "1w"
context.frequency = "30m"
# 设置回测基准, 比特币："huobi_cny_btc", 莱特币："huobi_cny_ltc", 以太坊："huobi_cny_eth"
context.benchmark = "huobi_cny_btc"
# 设置回测标的, 比特币："huobi_cny_btc", 莱特币："huobi_cny_ltc", 以太坊："huobi_cny_eth"
context.security = "huobi_cny_btc"
# 设置策略的参数
# 快线回看周期为9
context.user_data.fastk_period = 9
# 慢线回看周期为3
context.user_data.slowk_period = 3
context.user_data.slowd_period = 3
# 使用简单平均
context.user_data.slowk_matype = 0
# 每个frequency读取历史数据的bar数
context.user_data.longest_history = 100
# 超买信号线
context.user_data.over_buy_signal = 80
# 超卖信号线
context.user_data.over_sell_signal = 20
# 至此initialize函数定义完毕。
# 阅读3，策略核心逻辑：
# handle_data函数定义了策略的执行逻辑，按照frequency生成的bar依次读取并执行策略逻辑，直至程序结束。
# handle_data和bar的详细说明，请参考新手学堂的解释文档。
def handle_data(context):
# 获取历史数据, 取后longest_history根bar
hist = context.data.get_price(context.security, count=context.user_data.longest_history, frequency=context.frequency)
if len(hist.index) < context.user_data.longest_history:
context.log.warn("bar的数量不足, 等待下一根bar...")
return
# 最高价
high_prices = np.array(hist["high"])
# 最低价
low_prices = np.array(hist["low"])
# 收盘价
close_prices = np.array(hist["close"])
# matype: 0=SMA, 1=EMA, 2=WMA, 3=DEMA, 4=TEMA, 5=TRIMA, 6=KAMA, 7=MAMA, 8=T3 (Default=SMA)
# 用talib计算K，D两条线
K, D = talib.STOCH(high_prices, low_prices, close_prices, fastk_period=context.user_data.fastk_period, slowk_matype=context.user_data.slowk_matype, slowk_period=context.user_data.slowk_period, slowd_period=context.user_data.slowd_period)
current_k_value = K[-1]
current_d_value = D[-1]
previous_k_value = K[-2]
previous_d_value = D[-2]
long_signal_triggered = False
short_signal_triggered = False
context.log.info("当前 K = %s, D = %s; 前一周期 K = %s, D = %s" % (current_k_value, current_d_value, previous_k_value, previous_d_value))
# 生成买入/卖出信号
# 当D < 超卖线, K线和D线同时上升，且K线从下向上穿过D线时，买入
if current_d_value < context.user_data.over_sell_signal and current_d_value > previous_d_value and current_k_value > previous_k_value and previous_k_value < previous_d_value and current_k_value > current_d_value:
long_signal_triggered = True
# 当D > 超买线, K线和D线同时下降，且K线从上向下穿过D线时，卖出
elif current_d_value > context.user_data.over_buy_signal and current_d_value < previous_d_value and current_k_value < previous_k_value and previous_k_value > previous_d_value and current_k_value < current_d_value:
short_signal_triggered = True
# 有卖出信号，卖出全部仓位
if short_signal_triggered:
if context.account.huobi_cny_btc >= HUOBI_CNY_BTC_MIN_ORDER_QUANTITY:
context.log.info("产生卖出信号")
# 若持有仓位，则全仓卖出
if context.account.huobi_cny_btc >= HUOBI_CNY_BTC_MIN_ORDER_QUANTITY:
context.log.info("正在卖出 %s" % context.security)
context.log.info("卖出数量为 %s" % context.account.huobi_cny_btc)
context.order.sell(context.security, quantity=str(context.account.huobi_cny_btc))
else:
context.log.info("仓位不足，无法卖出")
# 有买入信号，全仓买入
elif long_signal_triggered:
context.log.info("产生买入信号")
if context.account.huobi_cny_cash >= HUOBI_CNY_BTC_MIN_ORDER_CASH_AMOUNT:
context.log.info("正在买入 %s" % context.security)
context.log.info("下单金额为 %s 元" % context.account.huobi_cny_cash)
context.order.buy(context.security, cash_amount=str(context.account.huobi_cny_cash))
else:
context.log.info("现金不足，无法下单")
else:
context.log.info("无交易信号，进入下一根bar")