使用python和tushare,得到每日上涨家数,当上涨家数小于1700家,买入前3个月内最高点比最低点超过100%的,起始资金1000万,每次最多买5只,从今天开始回测三年,打印出结果和收益曲线

时间: 2024-03-23 11:39:12 浏览: 17
好的,我明白了。为了回答这个问题,我需要先确认您已经安装好了Python和tushare库。 以下是您所需要的代码: ```python import tushare as ts import pandas as pd # 设置tushare token ts.set_token('YOUR_TOKEN') # 初始化pro接口 pro = ts.pro_api() # 获取股票代码 def get_stock_codes(): data = pro.stock_basic(exchange='', list_status='L', fields='ts_code') return data['ts_code'].tolist() # 获取最近三个月的股票价格数据 def get_price_data(ts_code): today = pd.Timestamp.today() three_months_ago = today - pd.Timedelta(days=90) data = pro.daily(ts_code=ts_code, start_date=three_months_ago.strftime('%Y%m%d'), end_date=today.strftime('%Y%m%d')) return data # 计算股票的涨跌情况 def calculate_change(data): close = data['close'] pre_close = data['pre_close'] change = close - pre_close return (change > 0).sum() # 获取涨跌情况并筛选股票 def get_filtered_stocks(): stock_codes = get_stock_codes() filtered_stocks = [] for code in stock_codes: data = get_price_data(code) if len(data) == 0: continue high = data['high'].max() low = data['low'].min() if high / low > 2: change = calculate_change(data) if change < 1700: filtered_stocks.append(code) return filtered_stocks # 计算买入股票的数量和价格 def calculate_buying(stock_codes, funds): buying_stocks = [] for code in stock_codes: data = get_price_data(code) if len(data) == 0: continue close = data.iloc[-1]['close'] volume = funds // (close * 100 * 100) volume = max(min(volume, 500), 0) buying_stocks.append((code, volume, close)) return buying_stocks # 回测函数 def backtest(start_funds, years): funds = start_funds portfolio = [] for year in range(years): print('Year:', year+1) filtered_stocks = get_filtered_stocks() buying_stocks = calculate_buying(filtered_stocks, funds) if len(buying_stocks) == 0: print('No stocks to buy') continue print('Buying stocks:', buying_stocks) for stock in buying_stocks: funds -= stock[1] * stock[2] * 100 * 100 portfolio.append((stock[0], stock[1], stock[2])) if len(portfolio) > 5: sell = portfolio.pop(0) funds += sell[1] * stock[2] * 100 * 100 print('Selling stock:', sell) print('Portfolio:', portfolio) print('Funds:', funds) return portfolio, funds # 运行回测 portfolio, funds = backtest(10000000, 3) print('Final portfolio:', portfolio) print('Final funds:', funds) ``` 请注意,您需要将代码中的YOUR_TOKEN替换成您自己的tushare token。 代码中的get_stock_codes函数获取所有A股的股票代码列表,然后使用get_price_data函数获取每个股票最近三个月的价格数据。接下来,我们使用calculate_change函数计算每个股票的涨跌情况,并使用get_filtered_stocks函数筛选出符合条件的股票。 接下来,我们使用calculate_buying函数计算每个股票的买入数量和价格。注意,我们每次最多只能买入5只股票。 最后,我们使用backtest函数回测策略。在每一年中,我们首先获取符合条件的股票,然后计算每个股票的买入数量和价格。接下来,我们买入股票并更新我们的投资组合。如果我们的投资组合中已经有5只股票了,我们就需要卖掉其中一只股票,获得资金。最后,我们打印出投资组合和剩余资金。 请注意,代码中只是一个简单的示例,您需要根据自己的需求进行修改和优化。 如果您想打印出收益曲线,可以使用以下代码: ```python import matplotlib.pyplot as plt def calculate_portfolio_value(portfolio, funds, data): value = funds for stock in portfolio: code = stock[0] volume = stock[1] close = stock[2] value += volume * close * 100 * 100 return value def plot_portfolio_value(portfolio, funds): value_history = [] for year in range(3): filtered_stocks = get_filtered_stocks() buying_stocks = calculate_buying(filtered_stocks, funds) for stock in buying_stocks: portfolio.append(stock) if len(portfolio) > 5: portfolio.pop(0) funds = calculate_portfolio_value(portfolio, funds, []) value_history.append(funds) plt.plot(value_history) plt.show() plot_portfolio_value(portfolio, 10000000) ``` 这段代码会使用calculate_portfolio_value函数计算投资组合的价值,并使用plot_portfolio_value函数绘制出投资组合的价值随时间的变化。注意,这里的数据是随机生成的,与实际情况可能有所不同。

最新推荐

recommend-type

python射线法判断一个点在图形区域内外

主要为大家详细介绍了python射线法判断一个点在图形区域内外,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

python实现根据给定坐标点生成多边形mask的例子

今天小编就为大家分享一篇python实现根据给定坐标点生成多边形mask的例子,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
recommend-type

python计算波峰波谷值的方法(极值点)

主要介绍了python求极值点(波峰波谷)求极值点主要用到了scipy库,本文通过实例代码给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下
recommend-type

python实现简单点对点(p2p)聊天

主要为大家详细介绍了python实现简单点对点p2p聊天,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

python-opencv获取二值图像轮廓及中心点坐标的代码

今天小编就为大家分享一篇python-opencv获取二值图像轮廓及中心点坐标的代码,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
recommend-type

RTL8188FU-Linux-v5.7.4.2-36687.20200602.tar(20765).gz

REALTEK 8188FTV 8188eus 8188etv linux驱动程序稳定版本, 支持AP,STA 以及AP+STA 共存模式。 稳定支持linux4.0以上内核。
recommend-type

管理建模和仿真的文件

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

Redis验证与连接:安装成功验证技巧

![Redis从入门到精通](https://img-blog.csdnimg.cn/6f39d0a2adc3456b96e804eb45032480.png) # 1. Redis安装与验证** ### 2.2 Redis验证方法 Redis验证可以通过以下方法进行: - **命令行验证:** - 打开终端,输入 `redis-cli` 命令进入 Redis 命令行。 - 输入 `ping` 命令,如果返回 "PONG",则表示 Redis 正在运行。 - **客户端验证:** - 使用 Redis 客户端(如 RedisInsight)连接到 Redis 服务器。
recommend-type

3、自定义一个函数int compareStr(char *p1, char *p2),实现两个字符串的比较。相等返回0,大于返回1,小于返回0;编写主函数main()来调用自定义函数完成测试。

下面是自定义的 `compareStr` 函数的代码实现: ```c #include <stdio.h> int compareStr(char *p1, char *p2) { while (*p1 && *p2 && (*p1 == *p2)) { p1++; p2++; } return (*p1 > *p2) - (*p1 < *p2); } int main() { char str1[100], str2[100]; int result; printf("请输入第一个字符串:");
recommend-type

c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf

校园超市商品信息管理系统课程设计旨在帮助学生深入理解程序设计的基础知识,同时锻炼他们的实际操作能力。通过设计和实现一个校园超市商品信息管理系统,学生掌握了如何利用计算机科学与技术知识解决实际问题的能力。在课程设计过程中,学生需要对超市商品和销售员的关系进行有效管理,使系统功能更全面、实用,从而提高用户体验和便利性。 学生在课程设计过程中展现了积极的学习态度和纪律,没有缺勤情况,演示过程流畅且作品具有很强的使用价值。设计报告完整详细,展现了对问题的深入思考和解决能力。在答辩环节中,学生能够自信地回答问题,展示出扎实的专业知识和逻辑思维能力。教师对学生的表现予以肯定,认为学生在课程设计中表现出色,值得称赞。 整个课程设计过程包括平时成绩、报告成绩和演示与答辩成绩三个部分,其中平时表现占比20%,报告成绩占比40%,演示与答辩成绩占比40%。通过这三个部分的综合评定,最终为学生总成绩提供参考。总评分以百分制计算,全面评估学生在课程设计中的各项表现,最终为学生提供综合评价和反馈意见。 通过校园超市商品信息管理系统课程设计,学生不仅提升了对程序设计基础知识的理解与应用能力,同时也增强了团队协作和沟通能力。这一过程旨在培养学生综合运用技术解决问题的能力,为其未来的专业发展打下坚实基础。学生在进行校园超市商品信息管理系统课程设计过程中,不仅获得了理论知识的提升,同时也锻炼了实践能力和创新思维,为其未来的职业发展奠定了坚实基础。 校园超市商品信息管理系统课程设计的目的在于促进学生对程序设计基础知识的深入理解与掌握,同时培养学生解决实际问题的能力。通过对系统功能和用户需求的全面考量,学生设计了一个实用、高效的校园超市商品信息管理系统,为用户提供了更便捷、更高效的管理和使用体验。 综上所述,校园超市商品信息管理系统课程设计是一项旨在提升学生综合能力和实践技能的重要教学活动。通过此次设计,学生不仅深化了对程序设计基础知识的理解,还培养了解决实际问题的能力和团队合作精神。这一过程将为学生未来的专业发展提供坚实基础,使其在实际工作中能够胜任更多挑战。