有没有一种方法来表示复数来存储在Spark DF中？

from pyspark.sql import SparkSession
from pyspark.sql.types import StructType, StructField, DoubleType

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

# 定义Schema
schema = StructType([
    StructField("real", DoubleType(), nullable=False),
    StructField("imaginary", DoubleType(), nullable=False)
])

# 创建DataFrame
data = [(1.0, 2.0), (3.0, 4.0), (5.0, 6.0)]
df = spark.createDataFrame(data, schema)

# 显示DataFrame
df.show()