Я пытаюсь объединить логику для этого фрагмента кода в PySpark, чтобы уменьшить количество повторений кода. Любое руководство будет по достоинству оценено:
col0 = when(visit[1] == '0', to_date(visit[0])).otherwise(None)
col1 = when(visit[1] == '1', to_date(visit[0])).otherwise(None)
2 ответа
Вы можете сделать это, используя назначение деструктурирования и выражение генератора:
col0, col1 = (when(visit[1] == c, to_date(visit[0])).otherwise(None) for c in '01')
col = {i: when(visit[1] == f'{i}, to_date(visit[0])).otherwise(None) for col in (1,2)}
Примечание: для 2 столбцов вы получаете одну строку, но дополнительные затраты с точки зрения субъективной ясности кода значительно превышают выигрыш; предлагаемое решение, вероятно, стоит только для 3+ столбцов
Похожие вопросы
Новые вопросы
python
Python - это многопарадигмальный, динамически типизированный, многоцелевой язык программирования. Он разработан для быстрого изучения, понимания и использования, а также для обеспечения чистого и единообразного синтаксиса. Обратите внимание, что Python 2 официально не поддерживается с 01.01.2020. Тем не менее, для вопросов о Python, связанных с версией, добавьте тег [python-2.7] или [python-3.x]. При использовании варианта Python (например, Jython, PyPy) или библиотеки (например, Pandas и NumPy) включите его в теги.
En Español