pyspark中的when子句给出错误“未定义名称'when'”

Question

使用以下代码，我收到一条错误消息，未定义名称“ when”。

voter_df = voter_df.withColumn('random_val',
when(voter_df.TITLE == 'Councilmember', F.rand())
.when(voter_df.TITLE == 'Mayor', 2)
.otherwise(0))

为带有标题Councilmember的任何投票者的F.rand（）方法的结果添加一个名为random_val的列到voter_df。将市长的random_val设置为2。将其他任何标题设置为值0

Answer 1

第二个when语句是数据框的方法，但第一个when语句不是。

解决方案： 使用....'random_val',F.when(....