VSCode中使用Python pandas、datatable时能否实现列名自动补全？

配置得当的情况下完全可以实现这两种数据结构的列名自动补全，常用方案如下：

基础前提 #

• 首先确保VSCode已经安装微软官方的Python和Pylance扩展，所有补全能力都基于这两个基础扩展实现
• 在VSCode设置中将python.analysis.typeCheckingMode配置为basic及以上级别

方法1：静态类型标注（无需运行代码即可补全） #

pandas 1.3+和datatable都支持带类型的结构标注，你可以通过TypedDict提前定义列结构，示例如下：

from typing import TypedDict
import pandas as pd
import datatable as dt

# 定义列结构
class OrderData(TypedDict):
    order_id: str
    amount: float
    pay_time: str
    user_id: int

# pandas DataFrame标注
df_pd: pd.DataFrame[OrderData] = pd.read_csv("order.csv")
# datatable Frame标注
df_dt: dt.Frame[OrderData] = dt.fread("order.csv")

标注完成后，输入df_pd["、df_pd.或者df_dt[:, dt.f.的时候就会自动触发列名补全。

方法2：运行时动态补全（无需写类型标注） #

如果你不想额外写类型标注，可以通过以下两种方式触发动态补全：

• 使用VSCode的Jupyter交互窗口/Notebook模式，运行过DataFrame/Frame的创建代码后，变量已经加载到内存中，Pylance会自动读取已存在变量的列信息，直接触发补全
• 进入Debug模式，代码运行到对应数据变量已经被赋值的节点后，也会自动识别列名提供补全

方法3：专用增强插件 #

你可以在VSCode扩展商店直接搜索pandas补全相关的插件，安装后可以进一步提升无标注场景下的列名识别准确率，不需要额外复杂配置。

注意事项 #

• 普通.py文件未运行、也没有加类型标注的场景下，静态分析无法识别动态读取的文件的列名，不会触发补全
• datatable的列名补全能力略弱于pandas，优先推荐使用类型标注方案获得更稳定的补全效果

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