如何生成大小为 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