IF语句可以正确构建spark dataframe吗？

from pyspark.sql import SparkSession
from pyspark.sql.functions import when

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

# 创建示例DataFrame
data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
df = spark.createDataFrame(data, ["name", "age"])

# 使用IF语句进行条件筛选和转换
df_filtered = df.filter(df.age > 30)  # 筛选年龄大于30的数据行
df_transformed = df.withColumn("category", when(df.age > 30, "Senior").otherwise("Junior"))  # 添加新列并根据条件赋值

# 显示结果
df_filtered.show()
df_transformed.show()