对于可能复杂的数据类型,有没有比下面更优雅的方法来获得相应的真实numpy数据类型?
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
只要dtypes不是基类,就应该能够用数字参数(如0
(来调用它们。
import numpy as np
def dtype_to_real(rvs_dtype: np.dtype[Any]) -> np.dtype[Any]:
return rvs_dtype.type(0).real.dtype