n个色子的点数之和为m的概率计算-附python代码

分享一个计算若干色子出现某一个点数之和的代码：
例如：当色子个数为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