Python 用以一定概率选择的特定数字填充矩阵

我想用Python创建矩阵，即wxk的维数，其中w和k是参数。矩阵必须是这样的，即它只能用数字{1，…，4}填充，概率也可以作为参数提供，例如p_1=（我们选择“1”=1/3的概率，…，我们选择“4”=1/5的概率）。谢谢你的帮助。

也许你可以试试

numpy.random.choice（[1,2,3,4]，size=（w，k），p=[p1,p2,p3,p4]）

---

请查看。

生成具有定义概率的数字的简单解决方案 将numpy作为np导入

p1 = 0.25
p2 = 0.33
p3 = 0.25
p4 = 0.25

a = np.random.uniform()

p1 = float(input()) #get value of p1
print(a)

if a<=p1:
  print(1)
elif a<=p1+p2:
  print(2)
elif a<=p1+p2+p3:
  print(3)
elif a<=p1+p2+p3+p4:
  print(4)

p1=0.25
p2=0.33
p3=0.25
p4=0.25
a=np.random.uniform（）
p1=浮点（输入（））#获取p1的值
印刷品（a）
如果