如何让VS Code自动提示DataFrame的全部列名？

• 安装增强型Python代码提示插件
  直接在VS Code扩展商店搜索并安装Pylance或Pyright，这两款插件是官方推荐的Python语言服务工具，对pandas的类型推断支持远优于旧版的Jedi插件。

• 安装pandas类型存根包
  打开终端执行以下命令，安装pandas的类型存根文件，帮助编辑器更精准识别DataFrame的结构：

  pip install pandas-stubs
  

• 配置类型检查模式
  在VS Code设置中搜索python.analysis.typeCheckingMode，将其值设置为basic或strict，开启类型检查后，编辑器会更主动地分析DataFrame的列信息并给出提示。

• 给DataFrame添加明确的类型注解
  如果是静态定义的DataFrame，或者从文件读取的数据集，手动添加类型注解能大幅提升提示准确性。示例：

  from typing import TypedDict
  import pandas as pd
  
  # 定义DataFrame的列结构
  class SalesData(TypedDict):
      Country: str
      Product: str
      Price: float
      Qty: int
  
  # 给DataFrame绑定类型
  df: pd.DataFrame[SalesData] = pd.read_csv("sales_data.csv")
  

  这样编辑器就能准确识别df包含的所有列名，输入df.时会自动弹出列名提示。

• 生效验证
  完成上述操作后，重启VS Code或按下Ctrl+Shift+P选择"Reload Window"，再测试DataFrame的列名提示功能。

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