You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

在Rust中实现Scipy truncnorm功能:生成0-200截断正态随机数

Rust中生成0-200范围的截断正态分布随机数

不需要自行实现截断正态分布的生成逻辑,Rust生态里有现成的 crate 可以直接用,rand_distr 就是最常用的选择——它是 rand 生态的扩展库,提供了包括截断正态分布在内的多种概率分布实现,功能类似Scipy的truncnorm。

使用步骤

  1. 添加依赖
    在你的Cargo.toml中加入以下依赖:

    [dependencies]
    rand = "0.8"
    rand_distr = "0.4"
    
  2. 代码示例
    下面是生成0-200范围内截断正态分布随机数的具体代码:

    use rand::Rng;
    use rand_distr::{TruncatedNormal, Distribution};
    
    fn main() {
        // 初始化随机数生成器
        let mut rng = rand::thread_rng();
    
        // 定义截断正态分布的参数:均值、标准差、下限、上限
        // 这里以均值100,标准差30为例,可根据需求调整
        let dist = TruncatedNormal::new(100.0, 30.0, 0.0, 200.0)
            .expect("Invalid distribution parameters");
    
        // 生成10个符合要求的随机数
        for _ in 0..10 {
            let num = dist.sample(&mut rng);
            println!("{}", num);
        }
    }
    

关键说明

  • TruncatedNormal::new会校验参数合法性:均值必须在[下限, 上限]范围内,标准差必须为正,否则会返回错误,需用expect或其他方式处理错误。
  • 该 crate 内部封装了高效的截断采样逻辑,无需手动编写拒绝采样循环(先采样正态分布再过滤范围的方式效率低,且存在性能隐患)。

为什么之前的方法无效

你尝试的gen_range() + sample(Standard)组合,本质是把均匀分布和标准正态分布的结果强行组合,这无法得到真正的截断正态分布。截断正态分布的核心是在正态分布基础上只保留指定范围内的样本,rand_distr的实现才符合这个数学定义。

内容的提问来源于stack exchange,提问作者theloosygoose

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.25 05:42:10