如何使用 numpy 获取所有可能的随机组合

我有一个长度为 NumPy 的数组N含有X那些和N-X零。我想生成数组的所有可能的随机组合。

例如，当N是 5 并且X是2，我需要以下组合，

[0, 0, 1, 1, 0]
[1, 1, 0, 0, 0]
[0, 0, 0, 1, 1]
[0, 0, 1, 0, 1]
[1, 0, 0, 1, 0]
[0, 1, 1, 0, 0]
[0, 1, 0, 1, 0]
[1, 0, 1, 0, 0]
[0, 1, 0, 0, 1]
[1, 0, 0, 0, 1]

在 numpy 中有没有直接的方法可以做到这一点？

组合在 Python 标准库中实现，位于itertools。它挑选X列表中的元素N，所以你可以让它选择索引并将它们设置为1:

import numpy as np
from itertools import combinations

N = 5
X = 2

for combination in combinations(range(N), X):
    arr = np.zeros(N)
    arr[list(combination)] = 1
    print(arr)

Output:

[1. 1. 0. 0. 0.]
[1. 0. 1. 0. 0.]
[1. 0. 0. 1. 0.]
[1. 0. 0. 0. 1.]
[0. 1. 1. 0. 0.]
[0. 1. 0. 1. 0.]
[0. 1. 0. 0. 1.]
[0. 0. 1. 1. 0.]
[0. 0. 1. 0. 1.]
[0. 0. 0. 1. 1.]