Python中numpy.real处理不同类型复数数组的异常问题

核心原因：数组的数据类型（dtype）不同 #

• 数组a：是Numpy原生复数类型数组，dtype为np.complex64或np.complex128。这类数组由Numpy直接管理连续内存块，复数的实部、虚部以底层数值形式存储，因此a.real或np.real(a)能直接提取实部，打印时显示为[[1.+2.j]]（无括号）。
• 数组b：是object dtype数组，数组内每个元素都是Python原生的complex标量对象。Numpy对object数组仅做容器管理，不会解析内部元素的数值结构，因此调用.real或np.real()时无法识别元素的复数属性，直接返回原数组；打印时沿用Python复数标量的显示格式，即带括号的[[(1+2j)]]。

验证与解决方法 #

1. 验证dtype：
   执行print(a.dtype)会输出complex128（或类似复数类型），执行print(b.dtype)会输出object。
2. 提取b的实部：
   • 手动遍历元素提取：[[x.real for x in row] for row in b]
   • 高效转换方式：将数组转为Numpy原生复数类型后提取：np.array(b, dtype=np.complex128).real
   • 小数据场景可用np.vectorize（本质是循环，效率有限）：np.vectorize(lambda x: x.real)(b)

手动复现b的方式 #

直接创建数组时Numpy会自动推断为原生复数类型，要生成object dtype的复数数组，需显式指定dtype：

import numpy as np
b = np.array([[complex(1, 2)]], dtype=object)
print(b)  # 输出 [[(1+2j)]]
print(b.real)  # 输出仍为 [[(1+2j)]]

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