如何从Pyspark调用另一个函数中的函数

Question

我需要帮助如何调用TeamCt和SeniorCt函数，并使用Python Pyspark将两个总价值转换为def Main函数。

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

Answer 1

希望这行得通:

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