numpy根据概率生成随机数

可以使用numpy中的random模块来生成随机数。如果要根据给定的概率分布生成随机数，可以使用numpy.random.choice函数。

例如，假设有一个列表x=[0,1,2,3]，以及一个对应的概率分布p=[0.1, 0.3, 0.4, 0.2]，表示生成的随机数在0,1,2,3中出现的概率分别为0.1, 0.3, 0.4, 0.2，可以使用以下代码生成随机数：

import numpy as np

x = [0, 1, 2, 3]
p = [0.1, 0.3, 0.4, 0.2]

# 生成10个随机数
random_numbers = np.random.choice(x, size=10, p=p)
print(random_numbers)

这段代码将生成10个随机数，并根据给定的概率分布来确定每个随机数所属的类别。例如，可能生成的结果为[2 2 1 2 2 1 2 0 2 1]，表示生成的10个随机数中有4个属于2，3个属于1，1个属于0，2个属于3。