python - 生成具有特定比例的 0 和 1 的随机数组

Question

我想生成一个大小为 N 的随机数组，它只包含 0 和 1，我希望我的数组具有 0 和 1 之间的某个比率。例如，数组的 90% 为 1，其余 10% 为 0 (我希望这 90% 与整个数组一起是随机的)。

现在我有:

randomLabel = np.random.randint(2, size=numbers)

但我无法控制 0 和 1 之间的比例。

Answer 1

如果您想要精确的 1:9 比例:

nums = numpy.ones(1000)
nums[:100] = 0
numpy.random.shuffle(nums)

如果您想要独立的 10% 概率:

nums = numpy.random.choice([0, 1], size=1000, p=[.1, .9])

或

nums = (numpy.random.rand(1000) > 0.1).astype(int)