import scipy.stats
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
%config InlineBackend.figure_format = 'retina'
## 生成隨機(jī)數(shù)
np.random.seed(123)
np.random.random(5)
array([ 0.69646919, 0.28613933, 0.22685145, 0.55131477, 0.71946897])
np.random.randint(0,9,10)
#生成0-9的10個(gè)隨機(jī)整數(shù)
array([6, 1, 0, 1, 0, 0, 3, 4, 0, 0])
蒙特卡洛模擬求圓周率
num = 1000
x = np.random.random(num)
y = np.random.random(num)
pi = np.sum(x**2 + y**2 < 1) / num * 4
print('PI:', pi)
## 為什么呢？
PI: 3.196
plt.figure(figsize=(5,5))
plt.scatter(x,y, alpha = 0.6)
plt.axis([0,1,0,1])
x2 = np.arange(0, 1.01, 0.01)
y2 = np.sqrt(1 - x2**2)
plt.plot(x2, y2, 'm', lw=3)
plt.show()
output_7_0.png
連續(xù)分布和正太分布
模擬面包重量的分布
假設(shè)是均值為950克贴浙，標(biāo)準(zhǔn)差為50克的正態(tài)分布崎溃。
那個(gè)數(shù)學(xué)家的故事盯质，想起來(lái)了
mean = 950
std = 50
# 生成滿足正態(tài)分布的隨機(jī)數(shù)，并繪制直方圖
sample = np.random.normal(mean, std, size=365)
plt.hist(sample, bins=30, alpha=0.7, rwidth=0.9, normed=True)
plt.show()
output_9_0.png
mean = 950
std = 50
norm = scipy.stats.norm(mean, std)
# 用 `scipy.stats.norm` 生成正態(tài)分布赎瑰，
##這是新的方法咯餐曼？
x = np.arange(700, 1200, 1)
## 這里的1鲜漩，是什么參數(shù)
y = norm.pdf(x)
plt.plot(x, y)
plt.show()
### 繪制正太分布的概率密度函數(shù)
output_11_0.png
### 累計(jì)分布函數(shù)cdf瓶佳，對(duì)了到底這幾個(gè)pdf和cdf的英文全名是什么鞍运恰臂拓？
# probability density function
# cumulative distribution function
x = np.arange(700, 1200, 1)
y = norm.cdf(x)
plt.plot(x, y)
plt.show()
output_12_0.png
### 計(jì)算買到的面包小于1000克的概率
# 這個(gè)好像很重要呢
#繪制PDF曲線
x = np.arange(700, 1200, 1)
y = norm.pdf(x)
plt.plot(x, y)
#在1000處繪制豎線
plt.vlines(1000, 0, norm.pdf(1000))
#填充顏色
x2 = np.arange(700, 1000, 1)
y2 = norm.pdf(x2)
plt.fill_between(x2, y2, color='blue', alpha=0.1)
#設(shè)置y軸范圍
plt.ylim(0,0.0085)
plt.show()
norm.cdf(1000)
## 其實(shí)累計(jì)分布才能解決傻工，低于1000的問(wèn)題
0.84134474606854293
計(jì)算買到的面包大于1000克的概率
#繪制PDF曲線
x = np.arange(700, 1200, 1)
y = norm.pdf(x)
plt.plot(x, y)
#繪制豎線
plt.vlines(1000, 0, norm.pdf(1000))
#填充顏色
x2 = np.arange(1000, 1200, 1)
y2 = norm.pdf(x2)
plt.fill_between(x2, y2, color='blue', alpha=0.1)
#設(shè)置y軸范圍
plt.ylim(0,0.0085)
plt.show()
output_16_0.png
1 - norm.cdf(1000)
## 對(duì)呀，用1-另一半就可以了呀
0.15865525393145707
norm.sf(1000)
### 這個(gè)sf是什么意思
## 好像sf是反函數(shù)坊饶。但忘了 具體是反什么了匿级？
0.15865525393145707
## 計(jì)算面包在950-1050的概率
#繪制PDF曲線
x = np.arange(700, 1200, 1)
y = norm.pdf(x)
plt.plot(x, y)
#繪制豎線
plt.vlines(950, 0, norm.pdf(950))
plt.vlines(1050, 0, norm.pdf(1050))
#填充顏色
x2 = np.arange(950, 1050, 1)
y2 = norm.pdf(x2)
plt.fill_between(x2, y2, color='blue', alpha=0.1)
#設(shè)置y軸范圍
plt.ylim(0,0.0085)
plt.show()
output_19_0.png
norm.cdf(1050) - norm.cdf(950)
#cdf（1050）是指低于1050的函數(shù)的意思么痘绎？
0.47724986805182079
# 90%的情況下孤页，買到的面包是小于多少克的？
### ppf才是反函數(shù)吧蕉堰，就是要實(shí)現(xiàn)某概率，該是什么初始變量
norm.ppf(0.9)
1014.07757827723
# 80%的情況下冰寻，買到的面包是大于多少克的斩芭？
norm.isf(0.8)
907.91893832135429
離散分布
outcome = np.random.randint(0,2,10)
outcome
array([1, 0, 0, 1, 1, 0, 1, 1, 0, 1])
np.sum(outcome)
6
sample = [np.sum(np.random.randint(0,2,10)) for i in range(10000)]
sample = pd.Series(sample)
# pdserie 什么意思啊
sample.value_counts().sort_index().plot.bar()
plt.show()
output_28_0.png
# 投硬幣問(wèn)題的二項(xiàng)分布
n = 10
p = 0.5
binomial = scipy.stats.binom(n, p)
x = np.arange(0,11)
plt.plot(x, binomial.pmf(x), 'bo')
plt.vlines(x, 0, binomial.pmf(x), colors='b')
plt.ylim(0,0.3)
plt.show()
output_30_0.png
mean, var = binomial.stats()
print(mean)
print(var)
5.0
2.5
### 應(yīng)用
# 某家風(fēng)投企業(yè)，投資成功的概率是5%琴庵，如果它投了100個(gè)項(xiàng)目仰美，恰好有5個(gè)成功的概率是多少？
n = 100
p = 0.05
binom = scipy.stats.binom(n,p)
x = np.arange(0,101)
plt.plot(x, binom.pmf(x), 'bo')
plt.vlines(x, 0, binom.pmf(x), colors='b')
plt.ylim(0,0.2)
plt.show()
output_33_0.png
binom.pmf(5)
## pmf概率質(zhì)量函數(shù)
0.18001782727043672
1 - binom.cdf(4)
## 累計(jì)概率分布
0.56401869931428927
binom.sf(4)
0.56401869931429105
binom.isf(0.1)
#10%的情況下，能成功多少
8.0
binom.ppf(1 - 0.1)
8.0
## 離散分布 - 泊松分布
# 有一家便利店使用泊松分布來(lái)估計(jì)周五晚上到店買東西的顧客數(shù)懦尝，根據(jù)以往數(shù)據(jù)壤圃，周五晚上平均每個(gè)小時(shí)的顧客數(shù)是20埃唯。
lmd = 20
poisson = scipy.stats.poisson(lmd)
x = np.arange(0,40)
plt.plot(x, poisson.pmf(x), 'bo')
plt.vlines(x, 0, poisson.pmf(x), colors='b')
plt.ylim(0,0.1)
plt.show()
output_40_0.png
mean, var = poisson.stats()
print(mean)
print(var)
20.0
20.0
#顧客數(shù)恰好是20的概率？
poisson.pmf(20)
0.088835317392084806
# 顧客數(shù)小于15的函數(shù)
poisson.cdf(15)
0.15651313463974229
# 顧客數(shù)大于等于20的概率模蜡？
poisson.sf(19)
0.52974273316075782
# 百分之90%的情況下忍疾，顧客數(shù)不會(huì)超過(guò)多少卤妒？
poisson.ppf(0.9)
26.0
#作業(yè)## 基本作業(yè)
#機(jī)票超賣現(xiàn)象
#假設(shè)某國(guó)際航班有300個(gè)座位，乘客平均誤機(jī)率是2%共缕。
#1、如果一共賣出305張機(jī)票翩活，那么登機(jī)時(shí)人數(shù)超額的概率是多少菠镇？
# 某家風(fēng)投企業(yè)承璃，投資成功的概率是5%盔粹，如果它投了100個(gè)項(xiàng)目，恰好有5個(gè)成功的概率是多少玻佩？
n = 305
p = 0.98
binom = scipy.stats.binom(n,p)
binom.pmf(301)
0.12929463926438506
1 - binom.cdf(300)
0.26915013819815137
binom.pmf(301)
0.12929463926438506
n = 305
p = 0.98
binom = scipy.stats.binom(n,p)
lmd=300
poisson = scipy.stats.poisson(lmd)
poisson.ppf(0.9)
322.0
