是否可以在pyspark select dataframe中检查列是否存在？

from pyspark.sql import SparkSession

# 初始化 SparkSession
spark = SparkSession.builder \
    .appName("Check Column in DataFrame") \
    .getOrCreate()

# 示例 DataFrame
data = [("Alice", 1), ("Bob", 2), ("Cathy", 3)]
columns = ["Name", "Age"]

df = spark.createDataFrame(data, columns)

# 检查列是否存在
if "Age" in df.columns:
    df_selected = df.select("Name", "Age")
else:
    print("Column 'Age' does not exist")

# 展示选定的 DataFrame
df_selected.show()

# 停止 SparkSession
spark.stop()