python怎么按概率生成指定数字

801
2023/11/1 20:55:07
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要按概率生成指定数字,可以使用numpy库中的random.choice函数。这个函数可以根据给定的概率分布从指定的数字列表中选择一个数字。

下面是一个示例代码,演示如何使用random.choice函数按概率生成指定数字:

import numpy as np

# 定义数字列表和对应的概率分布
numbers = [1, 2, 3, 4, 5]
probabilities = [0.1, 0.2, 0.3, 0.2, 0.2]

# 使用random.choice函数生成指定数字
generated_number = np.random.choice(numbers, p=probabilities)
print(generated_number)

在这个示例中,数字列表为[1, 2, 3, 4, 5],对应的概率分布为[0.1, 0.2, 0.3, 0.2, 0.2]。通过调用np.random.choice函数,并传递numbers和probabilities参数,可以按照给定的概率生成一个数字。

这个示例代码的输出将根据概率分布而变化,可能输出1、2、3、4或5中的任意一个数字。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: python如何控制输出的小数位数