使用python和tushare,1.选取前三年内的任意一天,如果当日上涨家数小于1700家,则这一天作为trade_date,2.从trade_date往前推90天,选出这90天内最高价是最低价的2倍以上的股票作为股票池,3.在trade_date当日如果,股票池股票收十字线,则买入,持有3天则卖出,4.起始资金1000万,每次最多买5只,打印出结果和收益曲线

时间: 2024-03-24 08:37:15 浏览: 14
好的,我会根据您的要求进行回答。首先,需要安装tushare模块,可以使用以下命令: ```python !pip install tushare ``` 然后,需要获取股票数据。以下是获取tushare数据的代码: ```python import tushare as ts # 设置tushare token ts.set_token('your_token_here') # 初始化pro接口 pro = ts.pro_api() # 获取交易日历 trade_cal = pro.trade_cal(exchange='', start_date='20180101', end_date='20201231', fields='cal_date, is_open') # 获取前三年内的任意一天作为trade_date import random trade_date = random.choice(trade_cal[trade_cal['is_open'] == 1].cal_date.tolist()) print('选定的交易日为:', trade_date) # 获取trade_date前90天的数据 df = pro.daily(trade_date=trade_date, fields='ts_code, trade_date, open, high, low, close, pre_close, pct_chg') # 筛选出trade_date前90天最高价是最低价的2倍以上的股票作为股票池 df['spread'] = df['high'] / df['low'] stock_pool = df[df['trade_date'] < trade_date].groupby('ts_code').filter(lambda x: x['spread'].max() >= 2) print('股票池选取的股票数为:', len(stock_pool)) ``` 接下来,需要编写判断十字线的函数: ```python # 判断是否是十字星线 def is_doji(df): # 获取开盘价和收盘价 open_price = df.iloc[-2]['open'] close_price = df.iloc[-2]['close'] # 判断是否满足十字星线的条件 if abs(open_price - close_price) / (df.iloc[-3]['high'] - df.iloc[-3]['low']) <= 0.1 and \ min(open_price, close_price) > df.iloc[-3]['close']: return True else: return False ``` 然后,需要编写买卖股票的函数: ```python # 买卖股票的函数 def trade(stock_pool, money): # 获取当前时间的股票数据 df = pro.daily(trade_date=trade_date, ts_code=','.join(stock_pool['ts_code'].tolist()), fields='ts_code, trade_date, open, high, low, close, pre_close, pct_chg') # 筛选出当天收十字线的股票 doji_stocks = [] for ts_code in stock_pool['ts_code'].tolist(): sub_df = df[df['ts_code'] == ts_code] if is_doji(sub_df): doji_stocks.append(ts_code) # 如果没有股票收十字线,则不进行交易 if len(doji_stocks) == 0: return money # 计算每只股票的购买数量 num_per_stock = min(5, len(doji_stocks)) per_stock_money = money / num_per_stock num_per_stock = [int(per_stock_money / sub_df.iloc[-1]['close']) for ts_code, sub_df in df[df['ts_code'].isin(doji_stocks)].groupby('ts_code')] # 购买股票 for i, ts_code in enumerate(doji_stocks[:num_per_stock]): sub_df = df[df['ts_code'] == ts_code] buy_price = sub_df.iloc[-1]['close'] money -= num_per_stock[i] * buy_price # 持有3天后卖出 sell_date = pro.trade_cal(exchange='', start_date=trade_date, end_date='20201231', fields='cal_date, is_open') sell_date = sell_date[sell_date['is_open'] == 1].iloc[3]['cal_date'] df = pro.daily(trade_date=sell_date, ts_code=','.join(stock_pool['ts_code'].tolist()), fields='ts_code, trade_date, open, high, low, close, pre_close, pct_chg') sell_money = 0 for ts_code, num in zip(doji_stocks[:num_per_stock], num_per_stock): sub_df = df[df['ts_code'] == ts_code] sell_price = sub_df.iloc[-1]['close'] sell_money += num * sell_price return money + sell_money ``` 最后,可以运行主程序,计算收益曲线: ```python # 主程序 money = 100000000 profits = [] for i, row in trade_cal[trade_cal['cal_date'] >= trade_date].iterrows(): if row['is_open'] == 1: # 计算收益 money = trade(stock_pool, money) profit = (money - 100000000) / 100000000 profits.append(profit) print('交易日:{},当前资产:{},收益率:{}%'.format(row['cal_date'], money, round(profit * 100, 2))) # 绘制收益曲线 import matplotlib.pyplot as plt plt.plot(profits) plt.xlabel('交易日') plt.ylabel('收益率') plt.title('收益曲线') plt.show() ``` 希望对您有帮助!

相关推荐

最新推荐

recommend-type

Python实现的当前时间多加一天、一小时、一分钟操作示例

主要介绍了Python实现的当前时间多加一天、一小时、一分钟操作,结合实例形式分析了Python基于datetime模块进行日期时间操作相关使用技巧,需要的朋友可以参考下
recommend-type

Python编程判断这天是这一年第几天的方法示例

本文实例讲述了Python编程判断这天是这一年第几天的方法。分享给大家供大家参考,具体如下: 题目:输入某年某月某日,判断这一天是这一年的第几天? 实现代码: year=int(input('请输入年:')) month=int(input('请...
recommend-type

Python编程一级试卷1.docx

青少年编程能力等级测评线上测评服务平台是互联网技术、信息安全技术、人工智能技术的综合应用体现。目前,经过行业专家的指导把关,在专业团队的辛勤开发下,与测评项目配套的测评服务平台及信息化系统已经研发并...
recommend-type

在Python3.74+PyCharm2020.1 x64中安装使用Kivy的详细教程

主要介绍了在Python3.74+PyCharm2020.1 x64中安装使用Kivy的详细教程,本文通过图文实例相结合给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
recommend-type

python 使用pandas的dataframe一维数组和二维数组分别按行写入csv或excel

pandas写入excel 两种 一个是一行一行的写(本身写入时列)另一种是写入二维数组。 1.一行一行的写 i=1 loujian=2 dat=[i,loujian,ll,load_idx,bilv,eventype]#一个变量 print (dat) test = pd.DataFrame(data=dat)....
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

2. 通过python绘制y=e-xsin(2πx)图像

可以使用matplotlib库来绘制这个函数的图像。以下是一段示例代码: ```python import numpy as np import matplotlib.pyplot as plt def func(x): return np.exp(-x) * np.sin(2 * np.pi * x) x = np.linspace(0, 5, 500) y = func(x) plt.plot(x, y) plt.xlabel('x') plt.ylabel('y') plt.title('y = e^{-x} sin(2πx)') plt.show() ``` 运行这段
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。