以不等概率生成数字

Question

事情就是这样。

我想以不同的概率生成两个值，1 和 -1。 当我运行这个脚本时，我收到消息“choice() got an unexpected keyword argument 'p'”

谁能告诉我为什么会发生这种情况以及如何解决它？ 谢谢。

from scipy import *
import matplotlib.pyplot as plt
import random as r

def ruin_demo():


    tmax = 1000       #The number of game rounds
    Xi = 10           #The initial money the gambler has
    T = []
    M = []
    t = 1
    for N in linspace(0.3, 0.49, 100):     #The probability changing region

        meandeltaX = 1.0*N + (1-N)*(-1.0)  #The expected value for each game round
        M.append(meandeltaX)
        while (t < tmax and Xi > 0):

            deltaX  = r.choice((1, -1), p=[N, 1-N]) # The change of money during each game round
                #print deltaX
            Xi = Xi + deltaX
            t = t + 1
        T.append(t)
    plt.plot(M, T)
    plt.show()

ruin_demo()

Answer 1

您收到消息TypeError: choice() got an unexpected keyword argument 'p' 是因为random.choice（Python 标准库函数）没有关键字参数p。

可能你的意思是numpy.random.choice：

>>> numpy.random.choice((1,-1), p=[0.2, 0.8])
-1

您可以通过将import random as r 更改为import numpy.random as r 来修复您的代码，但是使用非标准的单字母名称来引用一个重要的模块只会让事情变得更加混乱。常规缩写是

import numpy as np

之后np.random.choice 工作。

---

现在诚然因为scipy 导入了numpy 的一部分，您可以通过它访问numpy.random，但是您的线路

from scipy import *   # <-- don't do this! 

应该避免。它使用行为不同的 numpy 版本破坏标准 Python 函数，并且在某些情况下给出相反的答案。