Python 排列组合

2020-08-19  本文已影响0人  一个扫地的垃圾

1.排列&组合

以上四种情况均为最常见的排列组合,从有无顺序和是否重复两个维度进行思考,建议理解并背诵。

2.使用Python表示排列组合

在使用python计算排列组合之前,需要计算阶乘,可以有两种方式,一是使用math库中的factorial函数,二是使用如下的递归函数。

def factorial(n):
    """
    计算阶乘
    :param n: data
    :return:
    """
    if n == 1:
        return 1
    else:
        return n * factorial(n-1)

按照排列的公式:

def my_permutation(n, k):
    """
    计算无重复有序的排列数
    :param n: 样本
    :param k: 抽取
    :return: 排列数
    """
    return factorial(n)//factorial(n-k)

按照组合的公式:

def my_combination(n, k):
    """
    计算无序且不重复的组合数
    :param n: 样本
    :param k: 抽取
    :return: 组合数
    """
    return factorial(n)//(factorial(n-k)*factorial(k))

3.例题

3.1 生日问题

假设一个班级中共有n个人,一年有365天,其中每天作为生日的概率是相等的,那么其中至少有两个人的生日在同一天的概率是多少?

此题属于全排列问题,需要反向思考,写出公式之后直接输入到python中计算

def birth(n):
    """
    计算至少有两人的生日在同一天的概率
    :param n: 班级人数
    :return: 概率
    """
    return 1 - factorial(365)/(factorial(365-n)*365**n)


# prob = birth(50)
# print("{:.2f}%".format(prob*100))

3.2 大乐透问题

大乐透的玩法是这样的,从35个红球中选择5个,从12个红球中选择2个,如果全中,那就中一等奖。那么请问,中一等奖的概率是多少?

此题属于组合问题,中奖的可能性为一种,因此分子为1,分母为所有的组合情况。

def prize():
    return 1/(my_combination(35, 12) * my_combination(12, 2))


prob = prize()
print("{:.15f}%".format(prob%100))
上一篇下一篇

猜你喜欢

热点阅读