[PYTHON] IF의 등가물 다음에 ELSE

IF의 등가물 다음에 ELSE

나는이 질문을 이전에 여기에서 보았고 나는 그것으로부터 교훈을 얻었다. 그러나 나는 그것이 작동해야한다고 생각할 때 나는 왜 내가 오류를 얻고 있는지 확신하지 못한다.

몇 가지 규칙에 따라 기존 Spark DataFrame에 새 열을 만들고 싶습니다. 여기에 내가 쓴 것이있다. iris_spark는 3 개의 별개 카테고리가있는 범주 형 변수 iris_spark가있는 데이터 프레임입니다.

from pyspark.sql import functions as F

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

다음 오류가 발생합니다.

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-157-21818c7dc060> in <module>()
----> 1 iris_spark_df=iris_spark.withColumn("Class",F.when(iris_spark.iris_class=='Iris-setosa',0,F.when(iris_spark.iris_class=='Iris-versicolor',1)))

TypeError: when() takes exactly 2 arguments (3 given)


---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-157-21818c7dc060> in <module>()
----> 1 iris_spark_df=iris_spark.withColumn("Class",F.when(iris_spark.iris_class=='Iris-setosa',0,F.when(iris_spark.iris_class=='Iris-versicolor',1)))

TypeError: when() takes exactly 2 arguments (3 given)

왜 그런가?

해결법

1. ==============================

   1.올바른 구조는 다음 중 하나입니다.

   올바른 구조는 다음 중 하나입니다.

   (when(col("iris_class") == 'Iris-setosa', 0)
   .when(col("iris_class") == 'Iris-versicolor', 1)
   .otherwise(2))
   

   이는

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

   또는:

   (when(col("iris_class") == 'Iris-setosa', 0)
       .otherwise(when(col("iris_class") == 'Iris-versicolor', 1)
           .otherwise(2)))
   

   이는 다음과 같습니다.

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

   일반 구문 :

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

   또는

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

   하이브 IF 조건문과 아마 섞일 것입니다.

   IF(condition, if-true, if-false)
   

   이 함수는 하이브를 지원하는 원시 SQL에서만 사용할 수 있습니다.

from https://stackoverflow.com/questions/39048229/spark-equivalent-of-if-then-else by cc-by-sa and MIT license