分享一个计算若干色子出现某一个点数之和的代码:
例如:当色子个数为2时,点数之和为2的概率就是1/36,和为1的概率为0
代码块
@author: ypfzhao
from numpy.polynomial.polynomial import Polynomial
def prob(n):
prob=dict()
p=Polynomial((0,1/6,1/6,1/6,1/6,1/6,1/6))**n
for i in range(n,n*6+1):
prob[i]=p.coef[i]
return prob
if __name__=='__main__':
while True:
n=input('Please input n:') #输入色子的个数
if n is 'q': #当输入q时,退出循环
break
m=input('Please input m:') #输入色子的点数之和
if m is 'q': #当输入q时,退出循环
break
try:
n=int(n) #将色子个数转换为整型数
m=int(m) #将色子点数转换为整型数
except:
continue
try:
print(m,prob(n)[m]) #输出点数出现的概率
except:
print(m,0) #点数不存在输出概率0