透過新增多欄或替換同名欄位,回傳新的 DataFrame。
語法
withColumns(*colsMap: Dict[str, Column])
參數
| 參數 | 類型 | 說明 |
|---|---|---|
colsMap |
dict | 欄位名稱與欄位的字詞。 目前僅支援一張地圖。 |
退貨
DataFrame: DataFrame 並新增或更換欄位。
Examples
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|
# +---+-----+----+----+