python - 获取模块的导入名称

Question

我有以下内容:

import numpy as np
v = np.ndarray([1,2,3,4,5])

我有兴趣检查 v 并以某种方式恢复它的类型 np.ndarray。

什么不起作用:

1. type(v).__name__ 给出不带 np 的 ndarray。
2. type(v).__qualname__ 给出不带 np 的 ndarray。
3. type(v).__module__ 给出 numpy 而不是 np。

如何取回np.ndarray？

Answer 1

一种方法是在 globals() 中查找模块使用的别名。

def name_of(x):
    t = type(x)
    alias = next(
        k
        for k, v in globals().items()
        if hasattr(v, "__name__") and v.__name__ == t.__module__
    )
    return f"{alias}.{t.__name__}"

使用示例:

>>> import numpy as np
>>> name_of(np.array([]))
'np.ndarray'

此外，由于字典迭代是按插入顺序进行的，因此这将返回首先声明的别名。