python - 如何在 VS Code 中启用 pyspark 和 numpy 的方法建议？

Question

我有一个看起来像这样的 python 文件

import pandas as pd
import numpy as np
import pyspark

df_pd = pd.DataFrame()

spark = pyspark.sql.SparkSession.builder.getOrCreate()
df_spark = spark.createDataFrame([
    [1, ]
], ["A"])

a = np.arange(10)

当我输入 df_pd. 并在输入点后等待时，VS Code 会给出诸如 join、pivot 等方法建议。

我希望 a 和 df_spark 具有相同的功能。但是，当我输入 a. 或 df_spark. 并等待时，不会显示任何方法建议。

有人知道如何使方法建议适用于 numpy 和 pyspark 吗？

其他信息:

• VS 代码版本:1.58.2
• Python 扩展版本:v2021.6.944021595
• pylance 扩展版本:v2021.7.4
• Visual Studio IntelliCode 扩展版本:v1.2.14
• 操作系统:macOS BigSurr 版本 11.1
• Numpy 和 pyspark 安装在所选解释器的环境中(与 pandas 安装的环境相同)

Answer 1

Pylance 和 Microsoft 语言服务器存在一些问题。他们不够聪明。您可以尝试一下 JediLSP 语言服务器吗？

"python.languageServer": "JediLSP",

更新:

您可以尝试使用“import pyspark.sql”或“from pyspark.sql import SparkSession”而不是“import pyspark”。

它是这样设计的。您可以引用this page获取开发者对此设计的解释。