如何将嵌套列添加到pyspark中的dataframe？

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

spark = SparkSession.builder.getOrCreate()

data = [("Alice", 25, "New York"), ("Bob", 30, "San Francisco")]
df = spark.createDataFrame(data, ["name", "age", "city"])

df_with_nested_column = df.withColumn("address", struct(df["city"]))

df_with_nested_columns = df.withColumn("address", struct(df["city"], df["state"]))

df_with_nested_column.show()

+-----+---+-------------+
| name|age|      address|
+-----+---+-------------+
|Alice| 25| [New York]  |
|  Bob| 30|[San Francisco]|
+-----+---+-------------+