如何用Python计算任意长度基向量和的所有可能组合？

算力阿峰

2026-8-9

解决思路

核心逻辑

你要的其实是所有非空基向量子集的和——每个基向量可以选择加入或不加入求和集合，但不能都不选，这样就能覆盖所有可能的组合结果（对应示例里的7种情况）。

步骤分解

生成基向量
对输入向量的每个元素，生成仅该位置保留原元素、其余位置为0的基向量。比如输入[a,b,c]，基向量就是[[a,0,0], [0,b,0], [0,0,c]]。
用Python实现的话可以用列表推导式：
```
vec = [a, b, c]
basis_vectors = [[0 if idx != i else val for idx, _ in enumerate(vec)] for i, val in enumerate(vec)]
```

生成所有非空子集并求和
不需要用permutations（它是生成排列，关注元素顺序，而我们要的是无顺序的子集选择），可以用以下两种方式：

方式一：用itertools.combinations遍历所有子集长度（从1到基向量总数），对每个组合内的向量逐位求和：

from itertools import combinations

result = []
# 遍历子集大小：1个基向量、2个基向量...直到所有基向量
for k in range(1, len(basis_vectors)+1):
    # 生成所有k个基向量的组合
    for combo in combinations(basis_vectors, k):
        # 逐位求和
        sum_vec = [sum(vec[idx] for vec in combo) for idx in range(len(vec))]
        result.append(sum_vec)

方式二：用itertools.product生成选择掩码（0代表不选对应基向量，1代表选），过滤掉全0的掩码后计算求和：

from itertools import product

n = len(vec)
# 生成所有0/1组合（共2^n种），排除全0的情况
masks = filter(lambda x: any(x), product([0,1], repeat=n))
result = []
for mask in masks:
    # 根据掩码选择基向量并求和
    sum_vec = [sum(basis_vectors[j][k] * mask[j] for j in range(n)) for k in range(n)]
    result.append(sum_vec)