如何在值为struct的情况下设置pyspark数据帧映射值？

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, struct

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

# 创建示例数据帧
data = [("Alice", 25, ("New York", "USA")),
        ("Bob", 30, ("London", "UK")),
        ("Charlie", 35, ("Sydney", "Australia"))]

df = spark.createDataFrame(data, ["Name", "Age", "Location"])

# 设置映射值
df = df.withColumn("City", col("Location").getField("city"))
df = df.withColumn("Country", col("Location").getField("country"))

# 显示结果
df.show()

+-------+---+----------------+---------+--------+
|   Name|Age|        Location|     City| Country|
+-------+---+----------------+---------+--------+
|  Alice| 25|[New York, USA]|New York|     USA|
|    Bob| 30|   [London, UK]|  London|      UK|
|Charlie| 35|[Sydney, Australia]|  Sydney|Australia|
+-------+---+----------------+---------+--------+