如何使用Tushare+Backtrader进行股票量化策略回测

转载 作者：撒哈拉 更新时间：2024-05-08 23:10:01 60 4

pip install tushare

import tushare as ts

# 初始化Tushare，替换YOUR_TOKEN为你的Token
ts.set_token('YOUR_TOKEN')

# 创建Tushare接口对象
pro = ts.pro_api()

# 调用Tushare接口函数，获取股票行情数据
data = pro.daily(ts_code='000001.SZ', start_date='20220101', end_date='20220131')

# 打印获取的数据
print(data)

import tushare as ts
# 获取沪深300指数从2014年1月1日到最新的行情数据
hs300 = ts.get_k_data('hs300', start='2014-01-01')

# 将RSI值添加到hs300数据中
hs300['RSI'] = RSI(hs300)

# 画图
plt.figure(figsize=(12,6))
plt.plot(hs300['date'], hs300['close'], label='Close') # 画出收盘价曲线
plt.legend(loc='upper left')
plt.twinx()
plt.plot(hs300['date'], hs300['RSI'], 'r', label='RSI') # 画出RSI曲线
plt.legend(loc='upper right')
plt.show()

import mplfinance as mpf

hs300 = hs300.set_index('date')
hs300.index = pd.to_datetime(hs300.index)
mpf.plot(hs300, type='candle', volume=True, mav=(5,10,20), figratio=(12,6), title='HS300 Candlestick Chart')

# Get data from tushare
df = ts.get_k_data('hs300', start='2014-01-01')
df['date'] = pd.to_datetime(df['date'])  # 将日期转换为datetime格式
df = df.set_index('date', drop=True)  # 将日期设置为索引
data = bt.feeds.PandasData(dataname=df, datetime=None,
                           open=0, high=1, low=2, close=3, volume=4, openinterest=-1)  # 创建数据源

# Define the strategy
class RSI(bt.Strategy):
    params = (('rsi_period', 14),)

    def __init__(self):
        self.rsi = bt.indicators.RSI(period=self.params.rsi_period)

    def next(self):
        if not self.position:
            if self.rsi < 30:
                self.buy(size=1)
        else:
            if self.rsi > 70:
                self.sell(size=1)

cerebro = bt.Cerebro()
cerebro.adddata(data)  # 添加数据源
cerebro.addstrategy(RSI)
cerebro.broker.setcash(1000000.0)  # 设置初始资金
cerebro.broker.setcommission(commission=0.001)  # 设置佣金
print('Starting Portfolio Value: %.2f' % cerebro.broker.getvalue())  # 输出初始资金
cerebro.run()  # 运行策略
print('Final Portfolio Value: %.2f' % cerebro.broker.getvalue())  # 输出最终资金
cerebro.plot()  # 绘制图表

60 4 0

• 三种缓存策略：Cache Aside 策略、Read/Write Through 策略、Write Back 策略

  作者：小林coding 计算机八股文网站：https://xiaolincoding.com 大家好，我是小林。 今天跟大家聊聊，常见的缓存更新策略。 Cache Aside（旁路缓存）策略； Rea

• git - Mercurial merge 策略 vs Git merge 策略

  我使用 git 多年，最近为了一个项目改用 mercurial。在过去的 6 个月里，我已经学会了如何通过命令行很好地使用 Mercurial。 这可能是我的想象，但在我看来，mercurial 在

• oauth-2.0 - Passport & JWT & Google/Facebook 策略 - 我如何结合 JWT 和 Google/Facebook 策略？

  这个问题适合任何熟悉的人 Node.js express Passport 带有 Passport 的 JWT 身份验证(JSON Web token ) Facebook OAuth2.0 或谷歌

• 记录平等的 Coq 策略？

  在 Coq 中，当试图证明记录的相等性时，是否有一种策略可以将其分解为所有字段的相等性？例如， Record R := {x:nat;y:nat}. Variables a b c d : nat.

• javascript Bootstrap 策略

  我正在处理的项目目前只有一个 Bootstrap 文件，用于初始化应用程序中的所有 javascript 对象。类似于下面的代码 if(document.getElementById('nav'))

• ubuntu - OpenLDAP 策略

  我正在考虑使用 OpenLDAP 在首次登录时添加密码到期和强制更改密码。 似乎使用 ppolicy 覆盖来实现这一点。 当我在 ppolicy.schema 中看到这个时，我开始使用 ppolicy

• 避免显卡敏感性的 WPF 策略？

  这基本上是我昨天问的一个问题的重新陈述，因为我得到的一个答案似乎没有理解我的问题，所以我一定是不清楚。我的错。 因为 WPF 依赖于 DirectX，所以它对卡和驱动程序的内部非常敏感。我有一个案例，

• SAML NameId 策略

  我是单点登录(SSO)概念的新手。我开始知道 SAML 请求和响应是实现 SSO 流程的最佳方式。然后我开始阅读有关 SAML2.0 的信息。我来了一个术语 NameIdPolicy 在 saml1.

• 新雪豹用户的 Perl 策略？

  关闭。这个问题需要更多 focused .它目前不接受答案。 想改进这个问题？更新问题，使其仅关注一个问题 editing this post . 5年前关闭。 Improve this questi

• .NET 混淆工具/策略

  关闭。这个问题是opinion-based 。目前不接受答案。 想要改进这个问题吗？更新问题，以便 editing this post 可以用事实和引文来回答它。 . 已关闭 9 年前。 Improv

• 用于限制数据库扩展的 Azure 策略

  在 Azure 上创建新的 SQL 数据库时，它将“计算+存储”选项设置为“2 vCore + 32GB 数据最大大小”作为默认配置，但我不想使用 vCore，我可以更改它。但问题是，是否可以通过策略

• Azure 策略 - 防止在未经身份验证的情况下创建应用程序服务

  我希望创建一项策略，防止在未启用身份验证的情况下创建应用服务(仅审核它们是不够的)。 以下策略可以正确识别未启用身份验证的现有资源: { "mode": "All", "policyRule"

• Azure 策略创建一个deployifnotexists 策略

  我正在尝试从现有 AuditIfNotExists 策略创建 DeployIfNotExists 策略。部署时不会出错，但会错误提示“没有相关资源与策略定义中的效果详细信息匹配”。当评估政策时。当我将

• Azure 策略创建一个deployifnotexists 策略

  我正在尝试从现有 AuditIfNotExists 策略创建 DeployIfNotExists 策略。部署时不会出错，但会错误提示“没有相关资源与策略定义中的效果详细信息匹配”。当评估政策时。当我将

• Django json 策略？

  我正在使用 wunderground 的 json api 来查询我网站上的天气状况。 api 为我提供了一个包含所有必要数据的漂亮 json 对象，但我每天只能进行多次调用。存储这些数据的首选方式是

• Java OOP 策略

  我有一个名为可视化数据结构的项目。我有这样的 OOP 设计。 Class VisualDataStructures extends JFrame Class ControlPanel extends

• javascript "include"策略

  这个问题在这里已经有了答案: 关闭 14 年前。 副本: Use javascript to inject script references as needed? Javascript 没有任何指

• Android StrictMode 策略

  Android 应用程序遇到了一些 ANR 问题，因此我实现了 StrictMode 策略。以前从未使用过这个，所以希望有人可以帮助解释以下内容: 为什么日志显示 2 个看似相似的违规行为，除了前 4

• algorithm - 寻找酒店谜语的算法/策略

  我目前正在尝试解决一个问题。假设我们在路上行驶，我们知道路上有 10 家酒店。每家酒店都有 0 到 6 星。我的问题是:找到选择星级酒店的最佳解决方案。唯一的问题是:您不能回头去参观您已经决定不去的酒

• 安卓 MVP 策略

  我正在将我的应用程序迁移到 MVP。从这个 konmik 中获得了有关静态演示者模式的提示 这是我的简要 MVP 策略。为简洁起见，删除了大部分样板和 MVP 监听器。这个策略帮助我改变了方向，证明了