python - 生成所有 n 选择 k 个二元向量 python

Question

是否有任何有效的方法(numpy 风格)来生成所有 n 选择 k 个二元向量(有 k 个)？例如，如果 n=3 和 k=2，那么我想生成 (1,1,0), (1,0,1)， (0,1,1)。

谢谢

Answer 1

我不知道这样效率如何，但这里有一个方法:

from itertools import combinations
import numpy as np

n, k = 5, 3


np.array(
    [
        [1 if i in comb else 0 for i in range(n)]
        for comb in combinations(np.arange(n), k)
    ]
)
>>>
array([[1., 1., 1., 0., 0.],
       [1., 1., 0., 1., 0.],
       [1., 1., 0., 0., 1.],
       [1., 0., 1., 1., 0.],
       [1., 0., 1., 0., 1.],
       [1., 0., 0., 1., 1.],
       [0., 1., 1., 1., 0.],
       [0., 1., 1., 0., 1.],
       [0., 1., 0., 1., 1.],
       [0., 0., 1., 1., 1.]])