如何从Pyspark调用其他函数中的函数

mbzjlibv  于 2023-01-16  发布在  Spark
关注(0)|答案(1)|浏览(169)

我需要帮助如何调用TeamCt和SeniorCt函数,并获得两个总值到def主函数使用Python Pyspark.

def Main(source1):

    
def TeamCt(source1):
    north_total = source1 
        .filter(F.col(‘team’) == ‘North’) 
        .groupBy(F.col(‘team’)) 
        .agg(F.count(‘*’).alias(‘n_total’)
    return north_total

def SeniorCt(source1):
    senior_total = source1 
        .filter(F.col(‘grade’) == ‘Senior’) 
        .groupBy(F.col(‘grade’)) 
        .agg(F.count(‘*’).alias(‘s_total’)
    return senior_total
fcwjkofz

fcwjkofz1#

希望这能奏效:

def Main(source1):
    north_total = TeamCt(source1)
    senior_total = SeniorCt(source1)

相关问题