您可以将struct函数与select一起使用，如下所示：

from pyspark.sql import functions as F
finalDF = df.select( "id",
                     F.struct("data.foo01", "data.foo02").alias("foo"),
                     F.struct("data.bar01", "data.bar02").alias("bar")
                     )
finalDF.printSchema

架构：

root
 |-- id: string (nullable = true)
 |-- foo: struct (nullable = false)
 |    |-- foo01: string (nullable = true)
 |    |-- foo02: string (nullable = true)
 |-- bar: struct (nullable = false)
 |    |-- bar01: string (nullable = true)
 |    |-- bar02: string (nullable = true)

您可以简单地使用struct pyspark函数。

from pyspark.sql.functions import struct
new_df = df.select(
  'id',
  struct('data.foo01', 'data.foo02').alias('foo'),
  struct('data.bar01', 'data.bar02').alias('bar'),
)

与struct pyspark函数相关的附加说明：它可以获取字符串列名列表，以便只将列移动到结构中，或者需要表达式列表。