Python除以numpy arrays时如何保持小数

Question

因此，我正在努力在 Python 中实现我自己的同质性统计测试版本，用户将提交列表列表，函数将计算相应的 chi 值。

我发现的一个问题是我的 function 在执行除法时删除了小数点，导致小样本量的 chi 值有些不准确。

这是代码：

import numpy as np
import scipy.stats as stats


def test_of_homo(list1):
    a = np.array(list1)
    #n = a.size
    num_rows = a.shape[0]
    num_cols = a.shape[1]
    dof = (num_cols-1)*(num_rows-1)
    column_totals = np.sum(a, axis=0)
    row_totals = np.sum(a, axis=1)
    n = sum(row_totals)
    b = np.array(list1)
    c = 0
    for x in range(num_rows):
      for y in range(num_cols):
        print("X is " + str(x))
        print("Y is " + str(y))
        print("a[x][y] is " + str(a[x][y]))
        print("row_totals[x] is " + str(row_totals[x]))
        print("column_total[y] is " + str(column_totals[y]))
        b[x][y] = (float(row_totals[x])*float(column_totals[y]))/float(n)
        print("b[x][y] is " + str(b[x][y]))
        numerator = ((a[x][y]) - b[x][y])**2
        chi =  float(numerator)/float(b[x][y])
        c = float(c)+ float(chi)
    print(b)
    print(c)
    print(stats.chi2.cdf(c, df=dof))
    print(1-(stats.chi2.cdf(c, df=dof)))

listc = [(21, 36, 30), (48, 26, 19)]

test_of_homo(listc)

打印结果时，我看到b[x][y]值为[[33 29 23] [35 32 25]]而不是33.35, 29.97, 23.68等。这导致我得到的 chi 值为 15.58 ap 为 0.0004 而不是预期的 14.5。

我试图将所有内容都转换为浮点数，但这似乎不起作用。 使用decimal.Decimal(b[x][y])导致类型错误。 有什么帮助吗？

Answer 1

我认为问题可能是由于您提供给列表中的 function 的号码所致。 请注意，如果您将列表转换为 Numpy 数组而不指定数据类型，它将尝试根据值进行猜测：

>>> listc = [(21, 36, 30), (48, 26, 19)]
>>> a = np.array(listc)
>>> a.dtype
dtype('int64')

以下是强制转换为所需数据类型的方法：

>>> a = np.array(listc, dtype=float)
>>> a.dtype
dtype('float64')

在您的 function 的第一行和第 9 行尝试，看看它是否解决了问题。 如果你这样做，你就不需要一直使用float() 。