Loops 如何在spark中创建循环？

嗨，我如何在spark中执行以下操作？我想使用下面的循环创建一个基于日期的权重表。请告诉我如何在spark中创建相同的表好吗

import datetime

from datetime import date
from dateutil.relativedelta import relativedelta

Last_month = datetime.datetime.strptime(todays_date,'%Y-%m-%d').date() + relativedelta(months=-1)
Last_month = Last_month.replace(day=1)
Last_month = Last_month.strftime("%Y-%m-%d")
Last_month_date = datetime.datetime.strptime(Last_month, "%Y-%m-%d").date()
print(Last_month)
print(todays_date)



def create_weights(Period):
    i=1
    DF = pd.DataFrame()
    while i <= 12:
        j = 1
        while j <= Period:
            ct = j - 11 - i
            nw_date = datetime.datetime.strptime(Last_month,'%Y-%m-%d').date() + relativedelta(months=ct)
            data = pd.DataFrame([[i, nw_date, j]], columns = ['period','mdate','weight'])
            j += 1
            DF = pd.concat([DF,data])
        i += 1
    return DF

weights = create_weights(12)
pd.set_option('display.max_rows', 144)
weights

导入日期时间
起始日期时间导入日期
从dateutil.relativedelta导入relativedelta
Last_month=datetime.datetime.StrTime（今天的_日期，%Y-%m-%d'）.date（）+relativedelta（月=-1）
上个月=上个月。替换（天=1）
上个月=上个月.strftime（“%Y-%m-%d”）
上个月日期=datetime.datetime.strtime（上个月，%Y-%m-%d”）.date（）
打印（上个月）
打印日期（今天）
def创建_权重（周期）：
i=1
DF=pd.DataFrame（）
虽然我需要添加一些示例数据和预期输出，但任何人都可以很容易地帮助您。我添加了一些附加代码，现在应该可以在python中工作了。应该有3列周期、日期和重量。你如何让它在spark中工作？