Я пытаюсь избавиться от этого выражения yeild и оператора '% g' и заменить его оператором return,...

Я выполняю задание в pyspark, где в какой-то момент я использую сгруппированный агрегат Pandas UDF