python - 在 PyCharm 中使用类型提示自动完成特定类型列表

Question

This answer展示了如何在 PyCharm 中为列表使用类型提示。但是是否可以提示 PyCharm 列表中存在哪些类型的对象？我知道 python 中的列表可以包含不同类型的对象。但只是为了自动完成，我想要这个。就像假设，如果我有一个 numpy.ndarrays 的 list，我可以提示它以便当我执行 obj_list[0]. 时，PyCharm为我提供 numpy.ndarray 的自动完成功能？谢谢

Answer 1

Pycharm 确实对此有一些支持，但目前还不完美(我使用的是 2018.2.4 版本)。

对于大多数类型，它运行良好(测试了内置类型和一些简单的自定义类):

# hint type in function document:

def func(a):
    """
    :param a:
    :type a list[str]
    :return:
    """
    a[0]   # will autocomplete 

# another way, use `typing` module
from typing import List
def func(a: List[str]):
    a[0]   # will autocomplete

不幸的是，这两种方法现在都不适用于 numpy 类型。 Pycharm 会提示 cannot find reference ndarray in __init__.py。