numpy 如何将条件语句与np.random.choice沿着使用

vltsax25  于 2023-03-30  发布在  其他
关注(0)|答案(1)|浏览(155)

我使用np.random.choice从2个提供的数组中创建2个均匀分布的样本:

arr1 = np.arange(1000, 5001, 500)
arr2 = np.arange(1000, 3001, 500)

op1 = np.random.choice(arr1, 100)
op2 = np.random.choice(arr2, 100)

然而,对于输出(op 1和op 2),我想强制一个条件,使op 1 [i]〉= op 2 [i]。有没有一种方法可以将这个条件与np.random.choice结合起来,所以我仍然得到op 1和op 2的随机均匀分布,op 1 [i]〉= op 2 [i]?谢谢!

ff29svar

ff29svar1#

检查是否所有索引都不满足该条件,如果不满足,则重新生成新的选择。

for i in range(100):
    while op1[i] < op2[i]:
        op2[i] = np.random.choice(arr2)

相关问题