Как вызвать ошибку для python create_map с неинициализированными ключами? - PullRequest
0 голосов
/ 04 декабря 2018

Я пытаюсь взять столбец из фрейма данных pySpark, используя карту из другого столбца, например отображать значения в фрейме данных из словаря, используя pyspark , но он выбрасывает нулевые значения для неинициализированных ключей

from pyspark.sql.functions import create_map, lit, col

mapping_expr = create_map([lit(x) for x in chain(*new_dict.items())])

df = df.withColumn("new_col", mapping_expr.getItem(col("old_col")))

однако это решение не является идеальным, поскольку оно не выдает исключение, когда нет неинициализированных значений.Это просто делает нулевые значения.Я хочу бросить исключение как можно скорее.Мне нужен эквивалент use strict; use warnings; для perl для этого случая.

Как мне установить этот код, чтобы он вызывал исключение для неинициализированного значения на этапе отображения?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...