如何在pyspark中解压list类型的列

from pyspark.sql.functions import explode

df = df.withColumn('unpacked_col', explode(df.list_col))

df = df.withColumn('unpacked_col', explode(df.list_col)).select('list_col', 'unpacked_col')

from pyspark.sql import SparkSession
from pyspark.sql.functions import explode

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

# 创建示例数据
data = [(1, [1, 2, 3]), (2, [4, 5]), (3, [6])]

# 创建DataFrame
df = spark.createDataFrame(data, ['id', 'list_col'])

# 解压list类型的列
df = df.withColumn('unpacked_col', explode(df.list_col))

# 显示结果
df.show()

+---+--------+------------+
| id|list_col|unpacked_col|
+---+--------+------------+
|  1|[1, 2, 3]|           1|
|  1|[1, 2, 3]|           2|
|  1|[1, 2, 3]|           3|
|  2|  [4, 5]|           4|
|  2|  [4, 5]|           5|
|  3|     [6]|           6|
+---+--------+------------+