python - 我应该如何为复杂的 dtype 获取相应的真实 numpy dtype？

Question

是否有比以下方法更优雅的方法来为可能复杂的 dtype 获取相应的真实 numpy dtype？

import numpy as np

def dtype_to_real(rvs_dtype: np.dtype[Any]) -> np.dtype[Any]:
    return np.empty(0, dtype=rvs_dtype).real.dtype

Answer 1

只要数据类型不是基类，您就应该能够使用数字参数调用它们，例如 0。

import numpy as np

def dtype_to_real(rvs_dtype: np.dtype[Any]) -> np.dtype[Any]:
    return rvs_dtype.type(0).real.dtype