DataFrame 열 조건부 연산

PySpark로 데이터 정제하기

Mike Metzger

Data Engineering Consultant

조건절

조건절은 다음과 같습니다:

.when(<if condition>, <then x>)

df.select(df.Name, df.Age, F.when(df.Age >= 18, "Adult"))

여러 번의 .when()

df.select(df.Name, df.Age, 
          .when(df.Age >= 18, "Adult")
          .when(df.Age < 18, "Minor"))

.otherwise()는 else와 같습니다

df.select(df.Name, df.Age,
          .when(df.Age >= 18, "Adult")
          .otherwise("Minor"))

PySpark로 데이터 정제하기