古典概型题目 python练习

例1. 

 袋中装有4只白球和2只红球.

         从袋中摸球两次,每次任取一球.有两种式:
         (a)放回抽样;    (b)不放回抽样.
求:    (1)两球颜色相同的概率;  

         (2)两球中至少有一只白球的概率.

import math
def func(n):#阶乘算法
    s = 1
    if n == 0:
        return s
    else:
        for i in range(1,n+1):#[1,n+1)
            s *= i
        return s

def C(n,m):
    return func(n)/(func(n-m)*func(m))
pa=C(4,1)*C(4,1)/(C(6,1)*C(6,1))+