当前位置: 代码迷 >> 综合 >> pycharm的type hint显示: pandas.DataFrame, pandas.Series numpy.ndarray 初步尝试
  详细解决方案

pycharm的type hint显示: pandas.DataFrame, pandas.Series numpy.ndarray 初步尝试

热度:59   发布时间:2023-12-21 13:37:49.0

问题:如何在pycharm上显示pandas.DataFrame, pandas.Series 或numpy.ndarray的类型注解?

解决方案

  • 先使用pip install data-science-types
import pandas, numpy# 函数foo的参数p1传?类型为pandas.DataFrame,参数p2传?类型为pandas.Series,
# 参数p3传?类型为numpy.ndarray
def foo(p1: pandas.DataFrame, p2: pandas.Series, p3: numpy.ndarray) -> None:print(type(p1))print(type(p2))print(type(p3))def main():a = pandas.DataFrame({
    'qw': [1, 2], 'xs': [3, 4]})b = pandas.Series([1, 2])c = numpy.ndarray([1, 3, 2])d = 1foo(a, b, c)# 将光标停留在foo上,pycharm的type hint会提示输入的参数类型有误foo(a, b, d)help(foo)if __name__ == '__main__':main()
  • pycharm 的type hint 提示
    在这里插入图片描述
  • 缺点:
    (1)需要额外的依赖包
    (2)不能检测出每一列的数据类型
  相关解决方案