python - 将 numpy 数据类型与字符串进行比较

Question

许多 numpy 函数将 dtype 参数作为字符串(如 "float64")或 numpy 数据类型(如 numpy.float64)甚至 python 数据类型(如 float )。

我需要比较两种数据类型并希望支持这种灵活的接口(interface)。是否存在所有这些形式都等效的函数？ IE。我想要最小的函数 f 这样

f("float64") == f(numpy.float64) == f(float)

numpy 内部使用什么？

Answer 1

您应该阅读 Scalars numpy 文档的页面，它描述了数据类型层次结构。

要比较数据类型本身，您可以使用 np.issubdtype .一些例子:

>>> import numpy as np
>>> np.issubdtype(np.int32, int)
True
>>> np.issubdtype(np.int32, float)
False
>>> np.issubdtype(float, np.floating)
True
>>> np.issubdtype(float, np.inexact)
True
>>> np.issubdtype(np.float32, float)
True
>>> np.issubdtype(np.float32, int)
False
>>> np.issubdtype(np.float32, np.floating)
True