编写一个结构列和int/string列都很少的spark dataframe？

from pyspark.sql import SparkSession
from pyspark.sql.types import StructType, StructField, IntegerType, StringType

spark = SparkSession.builder.appName("CreateDataFrame").getOrCreate()

schema = StructType([
    StructField("id", IntegerType(), True),
    StructField("name", StringType(), True)
])

data = [(1, "John"), (2, "Jane"), (3, "Alice")]
rows = [Row(id=row[0], name=row[1]) for row in data]

df = spark.createDataFrame(rows, schema)