如何用Python编写程序，计算集合中选取n/2个元素的组合数？

方法一：生成所有组合并统计数量 #

如果需要同时查看具体的组合内容和统计数量，直接用Python标准库的itertools.combinations即可，它能生成集合中选k个元素的所有不重复组合：

import itertools

# 目标集合
target_set = {0, 1, 2, 3}
# 选取的元素数量
select_count = 2

# 生成所有组合（返回迭代器，转成列表可查看具体内容）
all_combinations = list(itertools.combinations(target_set, select_count))
# 组合数就是列表的长度
combination_num = len(all_combinations)

print(f"所有组合：{all_combinations}")
print(f"组合数：{combination_num}")

运行结果：

所有组合：[(0, 1), (0, 2), (0, 3), (1, 2), (1, 3), (2, 3)]
组合数：6

注：combinations返回的是元组迭代器，转成列表就能直观看到所有组合，适合需要确认组合内容的场景。

方法二：直接用数学公式计算数值 #

如果只需要组合数的结果，不需要列出所有组合，用组合数公式C(n,k) = n!/(k!*(n-k)!)更高效，尤其是集合规模较大时：

高版本Python（3.10+） #

math库自带comb函数，直接调用即可：

import math

target_set = {0, 1, 2, 3}
select_count = 2

set_size = len(target_set)
combination_num = math.comb(set_size, select_count)

print(f"组合数：{combination_num}")

运行结果：6

兼容低版本Python #

如果你的Python版本低于3.10，可以自己实现组合数计算逻辑：

def calc_combination(n, k):
    # 边界情况处理
    if k < 0 or k > n:
        return 0
    if k == 0 or k == n:
        return 1
    # 利用组合数对称性减少计算量
    k = min(k, n - k)
    result = 1
    for i in range(1, k+1):
        result = result * (n - k + i) // i
    return result

target_set = {0, 1, 2, 3}
select_count = 2
set_size = len(target_set)
print(f"组合数：{calc_combination(set_size, select_count)}")

运行结果同样是6

总结 #

• 需要查看具体组合：用itertools.combinations
• 仅需组合数数值：用math.comb（高版本）或自定义函数（低版本）更高效

内容的提问来源于stack exchange，提问作者Ariel