python - 两个正数相乘在 Python 3 中给出负输出

Question

我有一个 DataFrame df1 :

df1.head() = 

             wght          num_links 
id_y  id_x                      
 3     133   0.000203          2      
       186   0.000203          2 
 5     6     0.000203          2      
       98    0.000203          2      
       184   0.000203          2

我需要计算一个名为 thr 的变量,

thr = N*(N-1)*2,

哪里N是 df1 的行数.

问题是当我计算 thr 时,Python 抛出一个负值(尽管所有输入都是正值):

ipdb> df1['wght'].count()*(df1['wght'].count()-1)*2
-712569744 

可能的提示

行数N是

ipdb> df1['wght'].count() 
137736 

因此，

ipdb> 137736*137735*2
37942135920.

考虑到可以分配给 int32 的最大值是2147483647 ，我怀疑 NumPy 考虑了 type(thr) = <int32> , 当它应该是 <int64> .这有意义吗？

请注意，我没有编写生成 df1 的代码因为

ipdb> df1['wght'].count() 
137736

但是，如果需要重现错误，请告诉我。

提前致谢。

Answer 1

您遇到了 np.int32 溢出，因此只需使用 len(df) 而不是 df.column.count()。

这是一个小演示:

In [149]: x = pd.DataFrame(np.random.randint(0,100,size=(137736, 3)), columns=list('ABC'))

In [150]: x.A.count() * (x.A.count() - 1) * 2
Out[150]: -712569744

In [151]: len(x) * (len(x) - 1) * 2
Out[151]: 37942135920

In [153]: type(x.A.count())
Out[153]: numpy.int32

In [154]: type(len(x))
Out[154]: int