withColumns
Renvoie un nouveau DataFrame en ajoutant plusieurs colonnes ou en remplaçant les colonnes existantes qui ont les mêmes noms.
Syntaxe
withColumns(*colsMap: Dict[str, Column])
parameter
parameter | Type | Description |
|---|---|---|
| dict | un dict de nom de colonne et Colonne. Actuellement, seule une seule carte est prise en charge. |
Renvoie
DataFrame: DataFrame avec des colonnes nouvelles ou remplacées.
Exemples
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|
# +---+-----+----+----+