在Python 3中使用numpy进行浮点数操作时，作者发现将float32数组求和后与标量相加会导致类型从float32变为float64。问题在于numpy对于标量和数组的类型提升规则不同。当涉及标量运算时，会检查标量值以确定结果类型，而数组运算则基于数组的dtype。这导致了在特定情况下，即使数组保持为float32，与标量相加也会提升到float64。解决方案是确保标量类型在运算前被显式转换为float32。
I want to perform some standard operations on numpy float32 arrays in python 3, however I'm seeing some strange behavior when working with numpy sum(). Here's an example session:
Python 3.6.1 |Anaconda 4.4.0 (x86_64)| (default, May 11 2017, 13:04:09)
[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)] on darwin
import numpy as np
np.__version__
Out[3]: '1.12.1'
a = np.ones(10).astype(np.float32)
np.sum(a).dtype
Out[5]: dtype('float32')
(np.sum(a)+1).dtype
Out[6]: dtype('float64')
(np.sum(a)+1.).dtype
Out[7]: dtype('float64')
(a+1).dtype
Out[8]: dtype('float32')
Any reason why adding a scalar to the result of a sum (which seems to have dtype of float32) would cast it back to float64? To be clear, I know that
最低0.47元/天 解锁文章
742
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
