使用pyspark将两个字符串列值连接到整数数组

from pyspark.sql import SparkSession
from pyspark.sql.functions import concat, array, lit

spark = SparkSession.builder.getOrCreate()

data = [("John", "Doe", 25), ("Jane", "Smith", 30)]
df = spark.createDataFrame(data, ["first_name", "last_name", "age"])

df = df.withColumn("full_name", concat(df.first_name, lit(" "), df.last_name))

df = df.withColumn("age_array", array(df.age))