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

Question

This answer 展示了如何在 PyCharm 中对列表使用类型提示。但是是否可以提示 PyCharm 列表中存在哪些类型的对象？我知道 python 中的列表可以有不同类型的对象。但只是为了自动完成，我想要这个。假设，如果我有一个list 或numpy.ndarrays，我可以提示一下，以便当我执行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。