PySpark使用withColumnRenamed重命名多个列

想使用spark withColumnRenamed函数同时更改两个列的名称，现在比较麻烦需要一对一的修改，想找到更便捷的方法。

data = sqlContext.createDataFrame([(1,2), (3,4)], ['x1', 'x2'])


data = (data


 .withColumnRenamed('x1','x3')


 .withColumnRenamed('x2', 'x4'))

1.使用toDF方法。

data.toDF('x3', 'x4')

2. 使用select重命名。

from pyspark.sql.functions import col



mapping = dict(zip(['x1', 'x2'], ['x3', 'x4']))


data.select([col(c).alias(mapping.get(c, c)) for c in data.columns])

3. 自己定义一个rename_columns函数。

def rename_columns(df, columns):


 if isinstance(columns, dict):


 for old_name, new_name in columns.items():


 df = df.withColumnRenamed(old_name, new_name)


 return df


 else:


 raise ValueError("'columns' should be a dict, like {'old_name_1':'new_name_1', 'old_name_2':'new_name_2'}")