NumPy中复数dtype与对应浮点dtype的互相获取方法

NumPy的dtype对象原生支持两类类型的对应属性查询，不需要手动维护映射表，直接调用即可满足需求。

1. 从指定复数类型获取对应浮点类型（自动推导fdtype） #

拿到目标复数网格Zs的dtype对象后，访问它的.real属性就能直接得到匹配的浮点类型，完全符合复数由两个同精度浮点数组成的规则：

import numpy as np

# 例1：Zs类型指定为np.csingle
z_dtype = np.csingle
fdtype = np.dtype(z_dtype).real
# 得到的fdtype为dtype('float32')，即np.single，符合预期

# 例2：Zs类型指定为complex128
z_dtype = np.dtype("complex128")
fdtype = np.dtype(z_dtype).real
# 得到的fdtype为dtype('float64')，匹配要求

原有生成数组的代码可以直接复用这个fdtype：

Rs = np.linspace(*realBounds, realResolution, dtype=fdtype)
Is = np.linspace(*imagBounds, imagResolution, dtype=fdtype)
# 后续构建Zs时直接指定目标z_dtype即可
Zs = Rs.reshape(-1, 1) + 1j * Is.reshape(1, -1)
Zs = Zs.astype(z_dtype, copy=False)

2. 反向映射：浮点类型转对应复数类型 #

访问浮点dtype对象的.complex属性即可直接得到对应的复数类型：

# 例1：np.single映射为np.csingle
f_dtype = np.single
z_dtype = np.dtype(f_dtype).complex
# 得到的z_dtype为dtype('complex64')，即np.csingle

# 例2：float64映射为complex128
f_dtype = np.dtype("float64")
z_dtype = np.dtype(f_dtype).complex
# 得到的z_dtype为dtype('complex128')

说明 #

该方法是NumPy dtype的原生特性，自动覆盖所有平台支持的内置浮点/复数精度对（float16↔complex32、float32↔complex64、float64↔complex128、float128↔complex256），不存在硬编码映射表的版本兼容问题。如果需要做类型校验，可以提前判断dtype的kind属性：复数类型的kind为'c'，浮点类型的kind为'f'，避免传入非法类型。

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