VSCode中PyArrow无自动补全/智能提示的解决方法咨询

我在VSCode中使用ms-python.python和ms-python.vscode-pylance扩展编辑以下PyArrow代码：

import pyarrow as pa

data = [
    pa.array([1, 2, 3, 4]),
    pa.array(['foo', 'bar', 'baz', None]),
    pa.array([True, None, False, True])
]

batch : pa.RecordBatch = pa.record_batch(data, names=['f0', 'f1', 'f2'])

目前无法获取方法、参数等自动补全/智能提示功能，推测原因是PyArrow的大部分实现为原生代码。以下是可行的解决方法：

1. 安装PyArrow官方存根 #

PyArrow 14.0.0及以上版本提供官方类型存根，直接安装对应包即可：

pip install pyarrow-stubs

安装后Pylance会自动识别存根，提供完整的类型提示与自动补全。若你的PyArrow版本低于14.0.0，建议先升级至最新稳定版再安装存根。

2. 配置Pylance增强类型推断 #

在VSCode的settings.json中添加以下配置，让Pylance对原生扩展模块启用更积极的类型推断：

{
    "python.analysis.typeCheckingMode": "basic",
    "python.analysis.useLibraryCodeForTypes": true,
    "python.analysis.inlayHints.functionReturnTypes": true,
    "python.analysis.inlayHints.variableTypes": true
}

其中python.analysis.useLibraryCodeForTypes会让Pylance尝试从库源码提取类型信息，对原生模块的提示效果有明显提升。

3. 第三方存根兼容旧版本 #

若无法升级PyArrow，可使用社区维护的types-pyarrow存根包：

pip install types-pyarrow

该包适配旧版PyArrow，能覆盖多数常用API的提示需求。

4. 手动添加类型注解辅助 #

在代码中显式添加类型注解，帮助Pylance识别变量类型，示例：

import pyarrow as pa
from pyarrow import RecordBatch, Array

data: list[Array] = [
    pa.array([1, 2, 3, 4]),
    pa.array(['foo', 'bar', 'baz', None]),
    pa.array([True, None, False, True])
]

batch: RecordBatch = pa.record_batch(data, names=['f0', 'f1', 'f2'])

显式注解能让Pylance更精准地提供补全和提示。

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