withColumns

透過新增多欄或替換同名欄位,回傳新的 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|
# +---+-----+----+----+