为何numpy中a-b与a+(-b)结果相同但数据类型不同？

问题复现 #

import numpy as np

minuend = np.array(1, dtype=np.int64)
subtrahend = 1 << 63

result_minus = minuend - subtrahend
result_plus = minuend + (-subtrahend)

print(result_minus == result_plus)  # True
print(type(result_minus))  # <class 'numpy.float64'>
print(type(result_plus))  # <class 'numpy.int64'>

差异原因 #

• subtrahend = 1 << 63是Python整数，值为9223372036854775808，超出了int64类型的最大值（2^63 - 1 = 9223372036854775807）。
• minuend - subtrahend的类型逻辑：NumPy二元运算的类型提升规则中，当int64数组与超出其范围的Python整数运算时，会将两者都提升为float64类型再计算——因为int64无法容纳subtrahend的取值，而float64可以近似表示这个大整数，最终结果类型为float64。
• minuend + (-subtrahend)的类型逻辑：先计算-subtrahend得到-9223372036854775808，这个值正好是int64类型的最小值（int64取值范围为[-2^63, 2^63-1]），属于int64的合法取值。此时NumPy会将该Python整数转换为int64类型，再与minuend（int64）做加法运算，结果保持int64类型。

技术说明查阅途径 #

可参考NumPy官方文档中的**类型提升（Type Promotion）**章节，重点关注数组与标量运算时的类型转换规则，以及整数类型溢出处理的相关内容。

内容的提问来源于stack exchange，提问作者bers