withColumnsRenamed

通过重命名多个列返回新的 DataFrame。 如果架构不包含给定的列名,则这是一个 no-op。

Syntax

withColumnsRenamed(colsMap: Dict[str, str])

参数

参数 类型 说明
colsMap dict 现有列名和相应的所需列名的听写。 目前仅支持单个映射。

退货

DataFrame:具有重命名列的数据帧。

示例

df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])

df.withColumnsRenamed({"age": "age2"}).show()
# +----+-----+
# |age2| name|
# +----+-----+
# |   2|Alice|
# |   5|  Bob|
# +----+-----+

df.withColumnsRenamed({"age": "age2", "name": "name2"}).show()
# +----+-----+
# |age2|name2|
# +----+-----+
# |   2|Alice|
# |   5|  Bob|
# +----+-----+

df.withColumnsRenamed({"non_existing": "new_name"}).show()
# +---+-----+
# |age| name|
# +---+-----+
# |  2|Alice|
# |  5|  Bob|
# +---+-----+