如何在 VS 代码中启用 pyspark 和 numpy 的方法建议？答案

【问题标题】：How to enable method suggestions for pyspark and numpy in VS code?如何在 VS 代码中启用 pyspark 和 numpy 的方法建议？
【发布时间】：2021-07-20 14:21:18
【问题描述】：

我得到了一个看起来像这样的 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 的环境相同）

【问题讨论】：

标签： python numpy visual-studio-code pyspark

【解决方案1】：

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

"python.languageServer": "JediLSP",

更新：

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

它是这样设计的。开发者对此设计的解释可以参考this page。

【讨论】：

试过了，没用。我有一个带有 jupyter 代码单元的 python 文件（每个销售都以 # %% 开头）。方法建议不适用于 py 文件中的 Microsoft、Pylance 或 JediLSP。但是，如果我运行我的代码单元格，vscode 会打开一个交互式窗口，在其底部我有一个部分，我可以在其中键入并在交互式 shell 中评估 python 代码。该方法建议适用于所有语言服务器
@zwithouta 感谢您的回复。我在 Github 上提交了一个错误。可以参考这个page参与讨论。
@zwithouta 你能尝试使用'import pyspark.sql'或'from pyspark.sql import SparkSession'而不是'import pyspark'吗？你能检查一下这个link
像魅力一样工作！ import pyspark.sql 足以获得spark 和df_spark 的建议。感谢您在 github 上打开问题！但是，我仍然不知道如何获得 numpy 数组工作的方法建议。我试图通过像这样from numpy import array 导入array，但它并没有解决问题
似乎是另一种问题，因为 numpy 建议在交互式窗口中也不起作用。但是 IntelliJ/Pycharm 会提出类型建议，所以这绝对是可能的。