python自动化交易学习笔记五-Demo8-策略已经设置了买入条件-设置卖出条件

from __future__ import (absolute_import, division, print_function,
                        unicode_literals)
import datetime  # 用于datetime对象操作
import os.path  # 用于管理路径
import sys  # 用于在argvTo[0]中找到脚本名称
import backtrader as bt # 引入backtrader框架

# 创建策略 begin
class TestStrategy(bt.Strategy):
    def log(self, txt, dt=None):
        ''' 策略的日志函数'''
        dt = dt or self.datas[0].datetime.date(0)
        print('%s, %s' % (dt.isoformat(), txt))
    def __init__(self):
        # 引用data[0]数据的收盘价数据
        self.dataclose = self.datas[0].close
        # 用于记录订单状态
        self.order = None
    def notify_order(self, order):
        if order.status in [order.Submitted, order.Accepted]:
            # 提交给代理或者由代理接收的买/卖订单 - 不做操作
            return
        # 检查订单是否执行完毕
        # 注意：如果没有足够资金，代理会拒绝订单
        if order.status in [order.Completed]:
            if order.isbuy():
                self.log('买开, %.2f' % order.executed.price)
            elif order.issell():
                self.log('卖平, %.2f' % order.executed.price)
            self.bar_executed = len(self)
        elif order.status in [order.Canceled, order.Margin, order.Rejected]:
            self.log('Order Canceled/Margin/Rejected')
        # 无等待处理订单
        self.order = None
    def next(self):
        # 日志输出收盘价数据
        self.log('Close, %.2f' % self.dataclose[0])
        # 检查是否有订单等待处理，如果是就不再进行其他下单
        if self.order:
            return
        # 检查是否已经进场
        if not self.position:
            # 还未进场，则只能进行买入
            # 当日收盘价小于前一日收盘价
            if self.dataclose[0] < self.dataclose[-1]:
                # 前一日收盘价小于前前日收盘价
                if self.dataclose[-1] < self.dataclose[-2]:
                    # 买买买
                    self.log('买开, %.2f' % self.dataclose[0])
                    # 记录订单避免二次下单
                    self.order = self.buy()
        # 如果已经在场内，则可以进行卖出操作
        else:
            # 卖卖卖
            if len(self) >= (self.bar_executed + 5):
                self.log('卖平, %.2f' % self.dataclose[0])
                # 记录订单避免二次下单
                self.order = self.sell()

# 创建策略 end

# 创建cerebro实体
cerebro = bt.Cerebro()
# 添加策略
cerebro.addstrategy(TestStrategy)
# 先找到脚本的位置，然后根据脚本与数据的相对路径关系找到数据位置
# 这样脚本从任意地方被调用，都可以正确地访问到数据
modpath = os.path.dirname(os.path.abspath(sys.argv[0]))
datapath = os.path.join(modpath, 'F:/GZH/自动化交易/历史数据/sh.600173history_k_data2021-12-31-2021-12-31.csv')
# 创建价格数据
data = bt.feeds.GenericCSVData(
    dataname = datapath,
    fromdate = datetime.datetime(2020, 1, 1),
    todate = datetime.datetime(2020, 2, 29),
    nullvalue = 0.0,
    dtformat = ('%Y/%m/%d'),
    datetime = 0,
    open = 1,
    high = 2,
    low = 3,
    close = 4,
    volume = 5,
    openinterest = -1
)
# 在Cerebro中添加价格数据
cerebro.adddata(data)
# 设置启动资金
cerebro.broker.setcash(100000.0)
# 打印开始信息
print('开始权益: %.2f' % cerebro.broker.getvalue())
# 遍历所有数据
cerebro.run()
# 打印最后结果
print('最终权益: %.2f' % cerebro.broker.getvalue())

　　v5　　\Python38\python.exe F:/test/test/demo8.py

开始权益: 100000.00
2020-01-02, Close, 5.21
2020-01-03, Close, 5.27
2020-01-06, Close, 5.23
2020-01-07, Close, 5.22
2020-01-07, 买开, 5.22
2020-01-08, 买开, 5.20
2020-01-08, Close, 5.08
2020-01-09, Close, 5.24
2020-01-10, Close, 5.21
2020-01-13, Close, 5.21
2020-01-14, Close, 5.17
2020-01-15, Close, 5.11
2020-01-15, 卖平, 5.11
2020-01-16, 卖平, 5.11
2020-01-16, Close, 5.06
2020-01-16, 买开, 5.06
2020-01-17, 买开, 5.07
2020-01-17, Close, 5.01
2020-01-20, Close, 4.99
2020-01-21, Close, 4.99
2020-01-22, Close, 4.99
2020-01-23, Close, 4.88
2020-02-03, Close, 4.39
2020-02-03, 卖平, 4.39
2020-02-04, 卖平, 4.22
2020-02-04, Close, 4.43
2020-02-05, Close, 4.43
2020-02-06, Close, 4.66
2020-02-07, Close, 4.73
2020-02-10, Close, 4.72
2020-02-11, Close, 4.70
2020-02-11, 买开, 4.70
2020-02-12, 买开, 4.70
2020-02-12, Close, 4.77
2020-02-13, Close, 4.68
2020-02-14, Close, 4.66
2020-02-17, Close, 4.75
2020-02-18, Close, 4.67
2020-02-19, Close, 4.64
2020-02-19, 卖平, 4.64
2020-02-20, 卖平, 4.64
2020-02-20, Close, 4.66
2020-02-21, Close, 4.77
2020-02-24, Close, 4.70
2020-02-25, Close, 4.73
2020-02-26, Close, 4.85
2020-02-27, Close, 4.86
2020-02-28, Close, 4.84
最终权益: 99999.00
 
Process finished with exit code 0

Demo7中加入卖出条件，还是在next方法中添加相关的代码。这需要明确退场条件已经现在是否有资金在市场内。

Strategy对象用position属性表示是否有资产在市场内
buy和sell方法会返回被创建的订单，此时订单尚未被执行
notify方法将用来提示订单状态的改变
卖出条件为在买入后的5天（即第6天）无论盈亏都卖出。这里需要注意如果没有资产在场内，只能进行买入，而不能进行卖出。
结果显示，这个策略在这段时间这只股票上测试结果是亏-1（实际手续费+进场价格滑点。成交价不一定）。先不在意这些，已经知道在哪里写入买入和卖出条件了，后面可以再更换条件，先流程过一变。。。
————————————————
参考：https://blog.csdn.net/m0_46603114/article/details/104972556