PySpark：从列值中命名一个新的数据框列

Question

我需要在我的数据框中创建新列，并从数据框值中命名这些列。事情是这样的：

new_df = df.withColumn(concat('col_',df['col1']) , df['col2']+df['col3']) 

但它告诉我不能使用列对象名称

Answer 1

让我知道如果这能帮助，

from pyspark.sql import functions as F 
new_df = df.withColumn('combcol',F.concat(F.lit('col_'),df['col1'])).groupby('col1').pivot('combcol').agg(F.first('col1')+F.first('col2'))