Jupyter Notebook中numpy共轭数幂运算结果异常问题咨询

问题复现 #

运行以下代码：

import numpy as np
c = 100
c_conj = np.conjugate(c)
print(c == c_conj)
print(c**5 == c_conj**5)

在Jupyter Notebook/JupyterLite中得到输出：

True
False

但在Google Colab等平台输出：

True
True

原因分析 #

这不是用户操作失误，核心差异来自np.conjugate返回值的类型差异：

• 在JupyterLite/部分本地Notebook环境中，np.conjugate(100)返回的是numpy复数标量（numpy.complex128），尽管其虚部为0。
  • 第一行c == c_conj返回True：numpy的比较运算符会自动兼容Python int和复数标量，仅比较实部是否相等。
  • 第二行c**5 == c_conj**5返回False：c**5是Python原生int（10000000000），而c_conj**5是复数标量（10000000000+0j），Python原生的相等判断不会自动将int与复数做兼容比较，即使实部完全一致也会返回False。
• 在Google Colab等环境中，np.conjugate(100)返回的是numpy整数标量（如numpy.int64），与Python int的幂运算结果比较时，numpy会自动做类型兼容处理，因此返回True。

可以通过添加类型打印验证差异：

print(type(c), type(c_conj))

JupyterLite中会输出类似 <class 'int'> <class 'numpy.complex128'>，Colab中则会输出 <class 'int'> <class 'numpy.int64'>。

解决方法 #

• 方法1：将共轭结果转为整数类型

  c_conj = np.conjugate(c).astype(int)
  

• 方法2：取共轭结果的实部（原数为实数，共轭等于自身）

  c_conj = np.real(np.conjugate(c))
  

• 方法3：统一使用numpy类型定义变量

  c = np.int64(100)
  

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