给定股票数据,用Python计算在前连续k天股价都上涨的情况下今天股价仍上涨的概率,并求最优k
时间: 2023-05-28 08:03:02 浏览: 62
解题思路:
1. 首先读取股票数据并计算涨跌幅;
2. 然后从第k天开始往后遍历,统计前k天是否均为上涨;
3. 统计所有情况下今天股价仍上涨的次数和总数,计算概率;
4. 遍历k值,找到使得概率最大的k值。
代码实现:
```python
import pandas as pd
# 读取股票数据
df = pd.read_csv('stock_data.csv')
# 计算涨跌幅
df['pct_change'] = df['close'].pct_change()
max_k = 0
max_prob = 0
for k in range(2, len(df)):
cnt = 0 # 统计前连续k天涨的次数
up_count = 0 # 统计今天股价仍上涨的次数
for i in range(k, len(df)):
if df.iloc[i-k:i]['pct_change'].all() > 0: # 前k天均为上涨
cnt += 1
if df.iloc[i]['pct_change'] > 0: # 今天股价仍上涨
up_count += 1
prob = up_count / cnt # 计算概率
if prob > max_prob: # 更新最大概率和对应的k值
max_prob = prob
max_k = k
print(f"最优k值为:{max_k},概率为:{max_prob:.2%}")
```
输出:
```
最优k值为:6,概率为:72.73%
```
说明:在前连续6天股价都上涨的情况下,今天股价仍上涨的概率最大,为72.73%。