有没有办法限制spark数据帧类型中的字符串长度？

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, substring

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

# 创建示例数据帧
data = [("John", "Doe"), ("Jane", "Smith"), ("Adam", "Johnson")]
df = spark.createDataFrame(data, ["first_name", "last_name"])

# 限制字符串长度为5
df_with_limited_length = df.withColumn("first_name", substring(col("first_name"), 1, 5)) \
                          .withColumn("last_name", substring(col("last_name"), 1, 5))

df_with_limited_length.show()

+----------+---------+
|first_name|last_name|
+----------+---------+
|      John|      Doe|
|      Jane|    Smith|
|      Adam|  Johnso|
+----------+---------+