在withColumn（“myColumn”，myEnum.Giraffe）中使用PySpark的“python枚举”的最佳方法

Question

我想这样做：

myEnum = Enum("myEnum", "Giraffe Crab Rat Human")

df_new = df.withColumn("AnimalEnumValueColumn", myEnum.Giraffe)

（我知道这是不正确的，但我想要一个具有相同预期效果的替代解决方案）

Answer 1

您需要使用.value从枚举对象获取值：

import pyspark.sql.functions as F

df_new = df.withColumn("AnimalEnumValueColumn", F.lit(myEnum.Giraffe.value))

有关更多详细信息，请参阅enum docs