如何有效地批量重命名pyspark中的列？

from pyspark.sql import SparkSession

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 读取数据
df = spark.read.csv("path/to/file.csv", header=True)

# 获取原始列名列表
old_columns = df.columns

# 定义新的列名列表
new_columns = ["new_" + col for col in old_columns]

# 批量重命名列
df_renamed = df
for i in range(len(old_columns)):
    df_renamed = df_renamed.withColumnRenamed(old_columns[i], new_columns[i])

# 显示重命名后的数据
df_renamed.show()