Когда предложение в 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 - это метод фрейма данных, но первый оператор when - нет.

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

Answer 2

использовать из импорта pyspark.sql.functions, когда