用Python编写一个组合数函数并调用

以下是用Python编写一个组合数函数的示例代码：

def combination(n, k):
    """
    计算组合数C(n, k)
    """
    if k == 0 or k == n:
        return 1
    if k > n or k < 0:
        return 0
    numerator = 1
    denominator = 1
    for i in range(k):
        numerator *= (n - i)
        denominator *= (i + 1)
    return numerator // denominator

# 调用组合数函数
print(combination(5, 2))  # 输出10
print(combination(10, 3))  # 输出120

在函数中，我们首先判断k是否等于0或n，如果是，直接返回1。然后判断k是否大于n或小于0，如果是，返回0。接着计算组合数公式，使用一个循环来计算分子和分母，最后返回分子除以分母的值。在调用函数时，我们可以传入n和k的值，函数会返回组合数的结果，我们可以将其打印出来。