如何让R语言中Box-Muller函数输出长度为n的向量

Box-Muller变换的核心特性是每次运算生成2个独立的标准正态分布随机数，所以默认实现会输出2n个样本。要得到长度为n的向量，可通过以下几种实用方式调整：

方法1：仅保留单次生成的其中一个样本 #

修改函数逻辑，在每次Box-Muller计算后只保留一个结果，丢弃另一个。循环n次即可得到n个样本，逻辑最简单直观。

示例代码（Python）：

import math
import random

def box_muller(n):
    samples = []
    for _ in range(n):
        u1 = random.uniform(0, 1)
        u2 = random.uniform(0, 1)
        z = math.sqrt(-2 * math.log(u1)) * math.cos(2 * math.pi * u2)
        samples.append(z)
    return samples

注：这种方法会浪费一半的随机数生成资源，适合对性能要求不高的场景。

方法2：根据n的奇偶性高效利用生成结果 #

充分利用Box-Muller每次生成两个样本的特性，计算需要运行的次数为ceil(n/2)，生成后截断到n个样本，避免资源浪费。

示例代码（Python）：

import math
import random

def box_muller(n):
    samples = []
    # 计算需要执行的Box-Muller次数：向上取整n/2
    iterations = (n + 1) // 2
    for _ in range(iterations):
        u1 = random.uniform(0, 1)
        u2 = random.uniform(0, 1)
        z1 = math.sqrt(-2 * math.log(u1)) * math.cos(2 * math.pi * u2)
        z2 = math.sqrt(-2 * math.log(u1)) * math.sin(2 * math.pi * u2)
        samples.extend([z1, z2])
    # 截断至目标长度n
    return samples[:n]

这种方法性能最优，适合对效率有要求的场景。

方法3：直接截断现有函数的输出 #

如果不想修改原有函数，可直接对生成的2n长度向量做切片，取前n个（或任意n个）结果：

# 假设your_box_muller_function是你已实现的函数，输入n输出2n长度向量
original_samples = your_box_muller_function(n)
n_length_samples = original_samples[:n]

适合快速临时调整的场景，但同样存在资源浪费。

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