如何在Python中将spark dataframe中的所有列值连接成一个字符串？

from pyspark.sql import SparkSession
from pyspark.sql.functions import concat_ws

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

# 创建示例DataFrame
data = [("John", 25, "USA"), ("Alice", 30, "Canada"), ("Bob", 35, "UK")]
df = spark.createDataFrame(data, ["name", "age", "country"])

# 使用concat_ws函数将所有列值连接成一个字符串
df_concatenated = df.withColumn("concatenated", concat_ws(",", *df.columns))

# 显示结果
df_concatenated.show(truncate=False)

+-----+---+-------+------------------+
|name |age|country|concatenated      |
+-----+---+-------+------------------+
|John |25 |USA    |John,25,USA       |
|Alice|30 |Canada |Alice,30,Canada   |
|Bob  |35 |UK     |Bob,35,UK         |
+-----+---+-------+------------------+