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'))