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

Question

我想在Python中创建矩阵，即w x k的维，其中w和k是参数。矩阵必须使得它仅用数字{1，...，4}填充，并且我们也可以将其提供为参数，例如p_1 =（选择“ 1”的概率= 1/3）。 ..，我们选择“ 4” = 1/5）的概率。谢谢您的帮助。

Answer 1

也许您可以尝试numpy.random.choice([1, 2, 3, 4], size=(w, k), p=[p1, p2, p3, p4])

请签出docs。

Answer 2

生成具有确定概率的数字的简单解决方案 将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)