Python自定义随机变量U(t)代码语法错误排查及实现咨询

1. 语法错误根源 #

你代码里的else z[:i] > c:是错误写法——Python中else不能直接跟随条件判断，需用elif来承接额外分支。
另外，z[:i] < c会返回一个布尔数组，直接放入if判断会触发ValueError，因为if需要单个布尔值而非数组。

2. 修正后的循环写法 #

如果坚持用循环实现，要逐个遍历元素索引而非切片：

from scipy.stats import norm
import numpy as np

c = 1.57
z = np.random.normal(0, 1, 100)
u = np.zeros_like(z)  # 创建与z同形状的零数组，比固定[1,100]更灵活

for i in range(len(z)):
    if z[i] <= c:
        u[i] = 0
    else:
        u[i] = norm.cdf(z[i])

3. 更高效的numpy向量化实现 #

numpy的核心优势是向量化操作，无需循环即可快速完成，代码更简洁：

from scipy.stats import norm
import numpy as np

c = 1.57
z = np.random.normal(0, 1, 100)
# 先计算所有元素的cdf，再将z<=c的位置重置为0
u = norm.cdf(z)
u[z <= c] = 0

也可以用np.where一步完成：

u = np.where(z <= c, 0, norm.cdf(z))

这类向量化方法在样本量较大时，比循环效率高出数倍。

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