python量化选股策略代码
时间: 2023-11-12 22:59:04 浏览: 109
量化策略代码,量化策略代码太难写啦,Python
5星 · 资源好评率100%
好的,以下是一个简单的 Python 量化选股策略代码示例:
```python
import tushare as ts
# 获取股票数据
df = ts.get_hist_data('600519', start='2020-01-01', end='2021-01-01')
# 计算均线
df['ma5'] = df['close'].rolling(5).mean()
df['ma10'] = df['close'].rolling(10).mean()
# 判断买入卖出信号
df['signal'] = 0
df.loc[df['ma5'] > df['ma10'], 'signal'] = 1
df.loc[df['ma5'] < df['ma10'], 'signal'] = -1
# 计算收益率
df['pct_change'] = df['close'].pct_change()
df['strategy_return'] = df['signal'].shift(1) * df['pct_change']
# 计算累计收益率
df['cum_return'] = (1 + df['strategy_return']).cumprod()
# 输出结果
print(df.tail())
```
这个策略的思路是:当短期均线(5日均线)上穿长期均线(10日均线)时,买入股票;当短期均线下穿长期均线时,卖出股票。这个策略比较简单,实际应用中需要根据具体情况进行调整和优化。
阅读全文