如何让VS Code识别awswrangler返回的Pandas DataFrame并显示帮助弹窗

这个问题的核心是VS Code搭载的Pylance语言服务无法自动推断wr.athena.read_sql_query方法的返回值类型，所以没法触发对应的Pandas DataFrame补全提示，通过手动标注类型即可解决。

• 方法1：Python 3.6及以上版本支持行内类型注解，直接给变量标注pd.DataFrame类型即可

import boto3
import awswrangler as wr
import pandas as pd

df1 = pd.DataFrame({"a":[1]})
df2: pd.DataFrame = wr.athena.read_sql_query(sql="...", database="...")

• 方法2：也可以使用注释式类型标注，兼容更低版本的Python写法

import boto3
import awswrangler as wr
import pandas as pd

df1 = pd.DataFrame({"a":[1]})
df2 = wr.athena.read_sql_query(sql="...", database="...")  # type: pd.DataFrame

如果标注后仍未生效，可以检查两个配置项：

• 确认VS Code当前选择的Python解释器对应你安装了pandas、awswrangler的运行环境
• 确认Pylance扩展已启用，类型检查功能未完全关闭

内容的提问来源于stack exchange，提问作者user3914505