如何用Numpy一行代码生成指定阈值的随机二进制数组？

嘿，你的需求我完全get到了——要生成一个指定大小、包含特定比例1的0-1数组，还希望能写成一行代码对吧？你现有的代码逻辑没问题，但Numpy其实提供了更贴合场景、更简洁的工具函数，不用绕弯子做随机数比较和判断～

给你推荐两种非常合适的一行实现方式：

方法1：用np.random.binomial（最推荐） #

这是专门针对伯努利试验（每次输出0或1）的函数，完美匹配你的需求，代码简洁还高效：

import numpy as np
result = np.random.binomial(n=1, p=threshold, size=arr_size)

• n=1：表示每次是单次伯努利试验，结果只能是0或1
• p=threshold：就是你要的1的比例（“成功”概率）
• size=arr_size：直接指定生成数组的大小

如果需要固定随机种子保证结果可复现（比如调试时），推荐用Numpy的新随机数生成器API：

result = np.random.default_rng(seed=42).binomial(1, threshold, arr_size)

方法2：用np.random.choice（更直观） #

这个方法逻辑更直白，直接指定可选值和对应的概率分布：

result = np.random.choice([0, 1], size=arr_size, p=[1-threshold, threshold])

这里p参数里的两个值分别对应0和1出现的概率，和你的需求完全对应，可读性拉满。

对比你原来的代码，这两种方法都是一步到位生成目标数组，省去了先生成连续随机数再做比较的步骤，不仅代码更短，在数组规模较大时性能也会更优哦～

内容的提问来源于stack exchange，提问作者Eran Moshe