с колонкой

Возвращает новый кадр данных, добавив столбец или заменив существующий столбец с тем же именем.

Синтаксис

withColumn(colName: str, col: Column)

Параметры

Параметр	Тип	Описание
colName	str	строка, имя нового столбца.
col	колонна	Выражение столбца для нового столбца.

Возвраты

DataFrame: DataFrame с новым или замененным столбцом.

Примечания

Этот метод вводит проекцию внутри себя. Таким образом, вызывая его несколько раз, например, через циклы, чтобы добавить несколько столбцов, могут создавать большие планы, которые могут вызвать проблемы с производительностью и даже StackOverflowException. Чтобы избежать этого, используйте select одновременно несколько столбцов.

Примеры

df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
df.withColumn('age2', df.age + 2).show()
# +---+-----+----+
# |age| name|age2|
# +---+-----+----+
# |  2|Alice|   4|
# |  5|  Bob|   7|
# +---+-----+----+