6 карт (комбинаторика)

#from scipy.misc import comb # Это если comb() навернётся
'''
Из колоды в 52 карты наугад взяли 6 карт. Найдите вероятности событий:
- среди выбранных карт по три карты двух разных мастей
- среди выбранных карт не более двух бубновых карт
В ответе приведите разделенные пробелом обыкновенные дроби.
'''

def f(n):
    """Факториал"""
    result = 1
    for i in range(2, n+1):
        result *= i
    return result

def comb(n, k):
    """Сочетание из n по k"""
    combinations = f(n)//(f(k)*f(n-k))
    return combinations

def loc(n, k):
    """Данная функция находит размещение из n элементов по k"""
    location = f(n)//(f(n-k))
    return location

all = comb(52, 6)

print('Всего вариантов: ', all)

print('ЗАДАЧА 1')

mast = comb(13, 3) * comb(13, 3)
r = mast * 6

print('По три карты двух разных мастей: ', mast)
print('Благоприятные исходы (в 6 раз больше): ', r)
print('Вероятность: ', r / all)
print('ОТВЕТ: ', r, '/', all, "\n")

print('ЗАДАЧА 2')

a = comb(13, 0) * comb(39, 6)
b = comb(13, 1) * comb(39, 5)
c = comb(13, 2) * comb(39, 4)
sum = a + b + c
res = sum / all

print('Ноль карт буби: ', a)
print('Одна карта буби: ', b)
print('Две карты буби: ', c)
print('Благоприятные исходы (как сумма исходов с выпадением 0, 1 и 2 карт буби): ', sum)
print('Вероятность: ', res)
print('ОТВЕТ: ', sum, '/', all)