使用udf统计与pyspark dataframe中的某个值匹配的键值

from pyspark.sql import SparkSession
from pyspark.sql.functions import udf
from pyspark.sql.types import BooleanType

spark = SparkSession.builder.appName("UDF Example").getOrCreate()

data = [("John", 25), ("Alice", 30), ("Bob", 35)]
df = spark.createDataFrame(data, ["Name", "Age"])

def match_value(value):
    # 这里可以根据具体需求编写匹配逻辑
    if value == "Alice":
        return True
    else:
        return False

# 注册UDF函数
match_udf = udf(match_value, BooleanType())

df_filtered = df.filter(match_udf(df["Name"]))
count = df_filtered.count()