如何在 Apache Spark 中複製 IF then ELSE 語句？

IF Then ELSE 的Spark 等效項

在Spark 中，您可以使用when() 函數將條件表達式應用於列函數。此函數可讓您為不同的條件指定 true 和 false 值。

程式碼錯誤和解決方案

您的程式碼拋出錯誤，因為您錯誤地使用了 when()功能。 when() 的正確語法是：

when(condition, value).when(...)

或

when(condition, value).otherwise(...)

在您的程式碼中，您向when() 函數提供了三個參數，這是不正確的。要解決此問題，您需要如下重寫程式碼：

iris_spark_df = iris_spark.withColumn(
    "Class", 
    F.when(iris_spark.iris_class == 'Iris-setosa', 0)
    .when(iris_spark.iris_class == 'Iris-versicolor', 1)
    .otherwise(2)
)

等效SQL 表達式

Spark when() 函數相當於中的CASE語句SQL:

CASE
    WHEN (iris_class = 'Iris-setosa') THEN 0
    WHEN (iris_class = 'Iris-versicolor') THEN 1
    ELSE 2
END

以上是如何在 Apache Spark 中複製 IF then ELSE 語句？的詳細內容。更多資訊請關注PHP中文網其他相關文章！