Когда предложение в pyspark выдает ошибку «имя», когда «не определено»

Question

С приведенным ниже кодом я получаю сообщение об ошибке, имя «когда» не определено.

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

Добавьте в voter_df столбец с именем random_val с результатами метода F.rand () для любого избирателя с заголовком «Член Совета». Установите random_val в 2 для мэра. Установите для любого другого заголовка значение 0

Answer 1

Вторым оператором when является метод данных, но первым оператором not является.

Решение: используйте ....'random_val',F.when(....