Python: pytype 兼容的 type() 命令?
Python: pytype-compatible type() command?
是否有类似 type
的命令可以通过集合递归到 return pytype 样式的声明(如果存在)?我意识到具有异构元素的集合可能会有问题。
例如,我希望看到类似这样的输出:
>>> fancy_type({1: {"a": 2.2}})
dict[int, dict[str, float]]
据我所知,没有这样的功能。但是,按照您在问题中描述的那样编写函数似乎相当简单:
from typing import Iterable, Mapping
def pretty_type(obj):
obj_type = type(obj).__name__
if isinstance(obj, Mapping):
k, v = next(iter(obj.items()))
return '{}[{}, {}]'.format(obj_type, pretty_type(k), pretty_type(v))
elif isinstance(obj, Iterable) and not isinstance(obj, str):
return '{}[{}]'.format(obj_type, pretty_type(obj[0]))
else:
return obj_type
测试
>>> print(pretty_type(0))
int
>>> print(pretty_type({1:{"a":2.2}}))
dict[int, dict[str, float]]
>>> print(pretty_type([1, 2, 3]))
list[int]
>>> print(pretty_type([{1:'a'}, {2:'b'}]))
list[dict[int, str]]
>>>
上面的代码可以很容易地扩展以适应任何自定义 input/ouput。当然,代码显然假定元素之间是同质的。正如您在问题中指出的那样,异构元素会产生问题,在我看来,这实际上取决于一个人对如何处理此类情况的喜好。
是否有类似 type
的命令可以通过集合递归到 return pytype 样式的声明(如果存在)?我意识到具有异构元素的集合可能会有问题。
例如,我希望看到类似这样的输出:
>>> fancy_type({1: {"a": 2.2}})
dict[int, dict[str, float]]
据我所知,没有这样的功能。但是,按照您在问题中描述的那样编写函数似乎相当简单:
from typing import Iterable, Mapping
def pretty_type(obj):
obj_type = type(obj).__name__
if isinstance(obj, Mapping):
k, v = next(iter(obj.items()))
return '{}[{}, {}]'.format(obj_type, pretty_type(k), pretty_type(v))
elif isinstance(obj, Iterable) and not isinstance(obj, str):
return '{}[{}]'.format(obj_type, pretty_type(obj[0]))
else:
return obj_type
测试
>>> print(pretty_type(0))
int
>>> print(pretty_type({1:{"a":2.2}}))
dict[int, dict[str, float]]
>>> print(pretty_type([1, 2, 3]))
list[int]
>>> print(pretty_type([{1:'a'}, {2:'b'}]))
list[dict[int, str]]
>>>
上面的代码可以很容易地扩展以适应任何自定义 input/ouput。当然,代码显然假定元素之间是同质的。正如您在问题中指出的那样,异构元素会产生问题,在我看来,这实际上取决于一个人对如何处理此类情况的喜好。