np.random.choice 错误：概率之和不等于 1，但 print 说是吗？

Question

我在使用 numpy 的 random.choice 函数时遇到了一个小问题。 我给它一个列表（a），以及与这个列表相关的概率（p）（我试图生成随机文本，实现一个二元马尔可夫模型，概率在训练语料库上计算）。 问题是，它在程序中崩溃，告诉我概率总和不等于 1。 更大的问题是，sum(p) 总和为 1。

这是一个错误吗？ random.choice 和常规 sum 函数的总和是否不同？我错过了什么吗？

代码如下：

def randomBigram(self):
    doc = open(self.path+"/randomGenBi.txt", "wb")
    lettre = str(np.random.choice(self.letters.index))
    a = [elem for elem in self.probaBigram.index if elem[1] == lettre]
    p = [self.probaBigram[elem] for elem in self.probaBigram.index if elem[1] == lettre]
    random = np.random.choice(a, p=p)
    i = 0
    while i < 5000:
        lettre = str(np.random.choice(self.letters.index))
        print "lettre", lettre
        a = [elem for elem in self.probaBigram.index if elem[1] == lettre]
        p = [self.probaBigram[elem] for elem in self.probaBigram.index if elem[1] == lettre]
        if sum(p) != 1.0:  #debug
            print "somme sur p:", sum(p)
            print "not equal"
        else:
            print "equals one"
        random = np.random.choice(a, p=p)
        doc.write(random)

        i += 1

这是我的 shell 输出示例：

字母a p 的总和：1.0 不相等

我真的不明白...

欢迎任何帮助:)

谢谢！

杰西卡

Answer 1

变化：

print "somme sur p:", sum(p)

到

print "somme sur p:", repr(sum(p))
                      ^^^^^      ^

然后再试一次。 print 在打印项目之前隐式地将 str() 应用于项目，并且在 Python 的“足够老”版本中 str 将浮点数舍入到 12 个有效数字。有许多不等于 1 的浮点数将打印为1.0。但唯一一个其 repr 显示为 1.0 的浮点数正好是 1.0。

一旦您发现总和确实不是 1.0，请向我们展示它是什么，并提出一个新问题，了解下一步该做什么;-)