python二项分布的概率使用

# 导入相关模块

import numpy as np

import tushare as ts

import pandas as pd

from scipy import stats

# 设定好接口  注意这句不能照抄，需要输入自己的接口密匙

token = 'Your token'   # 输入你的接口密匙，获取方式及相关权限见Tushare官网。

pro = ts.pro_api(token)

# 获取数据

df = pro.daily(ts_code='000001.SZ')  # daily为tushare的股票日线数据接口。

df['trade_date'] = pd.to_datetime(df['trade_date']) 

df.set_index(['trade_date'], inplace=True)  # 将日期列作为行索引

df = df.sort_index()

ret = df.pct_chg['2020']

# 估算平安银行股价上涨的概率

p = len(ret[ret > 0]) / len(ret)

print(p)

# 估计十个交易日中，平安银行有六个交易日上涨的概率

prob = stats.binom.pmf(6,10,p)

print(prob)