python - 如何生成大小为 41003 的 0 和 1 数组，并且 1 的数量只能是 100？

Question

我想生成一个大小为 41003 且包含 0 和 1 的数组。但 1 的数量应该是 100。如何在 python 中做到这一点？

我尝试使用

生成

np.random.randint(2, size=41003) 

但是1的数量不是100个。

Answer 1

初始化一个包含 n=41003 个零的列表。然后使用 random.sample(range(n), k=100) 在该列表上生成 100 个不重复的随机索引。
最后，将具有这些索引的列表中的值设置为 1。

from random import sample

n = 41003
values = [0] * n
for i in sample(range(0, n), k=100):
    values[i] = 1

print(values.count(1)) # 100