com colunas

Retorna um novo DataFrame adicionando várias colunas ou substituindo as colunas existentes que possuem os mesmos nomes.

Sintaxe

withColumns(*colsMap: Dict[str, Column])

Parâmetros

Parâmetro	Tipo	Descrição
colsMap	dicionário	Um dicionário contendo o nome da coluna e o tipo de coluna. Atualmente, apenas um mapa é suportado.

Devoluções

DataFrame: DataFrame com colunas novas ou substituídas.

Exemplos

Python

df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
df.withColumns({'age2': df.age + 2, 'age3': df.age + 3}).show()
# +---+-----+----+----+
# |age| name|age2|age3|
# +---+-----+----+----+
# |  2|Alice|   4|   5|
# |  5|  Bob|   7|   8|
# +---+-----+----+----+