如何在NumPy中检查并选用最大的可用复数数据类型

这个需求我之前也碰到过！确实直接写try-except创建数组的写法看起来有点乱糟糟的，其实有几种更优雅的方式来判断这些复数类型是否是平台支持的「真实」类型，而不是dummy类型，进而选出精度最高的可用类型。

方法一：通过检查类型的字节大小（高效无异常） #

NumPy中，平台不支持的dummy类型对应的dtype对象，其itemsize属性会是0（因为没有实际的内存占用）。我们可以利用这一点来快速筛选：

import numpy as np

def get_largest_available_complex_dtype():
    # 按精度从高到低排序候选类型
    candidate_dtypes = [np.complex512, np.complex256, np.complex192, np.complex128]
    for dtype in candidate_dtypes:
        dt = np.dtype(dtype)
        if dt.itemsize > 0:
            return dtype
    # 兜底：complex128几乎在所有平台都支持
    return np.complex128

方法二：尝试实例化标量（更稳妥） #

如果担心字节大小的判断不够准确，也可以尝试直接创建该类型的标量——dummy类型无法被实例化，会抛出TypeError，我们通过捕获这个异常来跳过不可用的类型：

import numpy as np

def get_largest_available_complex_dtype():
    candidate_dtypes = [np.complex512, np.complex256, np.complex192, np.complex128]
    for dtype in candidate_dtypes:
        try:
            # 尝试创建一个该类型的空标量，开销极小
            dtype(0 + 0j)
            return dtype
        except TypeError:
            # 类型不可用，继续检查下一个
            continue
    return np.complex128

使用示例 #

调用这个函数就能直接拿到当前平台支持的最大精度复数类型：

largest_complex_dtype = get_largest_available_complex_dtype()
print(f"当前选用的最大可用复数类型：{largest_complex_dtype}")

需要注意的是，np.complex128是NumPy的基础复数类型，几乎在所有平台上都能正常使用，所以用它做兜底非常安全。

备注：内容来源于stack exchange，提问作者NYG