我有一个Spark数据框,看起来像这样,其中商品编号,国家/地区代码和日期的每种组合都包含一行,而该组合中存在金额值。此数据框中大约有40万行。
articlenumber countrycode date amount
--------------------------------------------------
4421-222-222 DE 2020-02-05 200
1234-567-890 EN 2019-05-23 42
1345-457-456 EN 2019-12-12 107
现在,我需要一个附加列“金额12M”,它根据以下规则为每一行计算其值:
在每一行中,“金额12M”应包含“金额”中所有值的总和,其中商品编号和国家/地区代码与该特定行中的值匹配,并且日期位于该行中日期之前的12个月之间
是否需要为尚无值的日期/国家/商品编号组合添加金额为0的行?
由于我不是编程专家(工程专业的学生),我需要一些帮助如何在处理该数据框的python脚本中实现这一目标。
感谢您对此提出任何想法。
答案 0 :(得分:0)
已编辑:
import pyspark.sql.functions as f
from pyspark.sql import Window
w = Window.partitionBy('articlenumber', 'countrycode').orderBy('date').orderBy('yearmonth').rangeBetween(-11, 0)
df.withColumn('yearmonth', f.expr('(year(date) - 2000) * 12 + month(date)')) \
.withColumn('amount 12M', f.sum('amount').over(w)) \
.orderBy('date').show(10, False)
+-------------+-----------+----------+------+---------+----------+
|articlenumber|countrycode|date |amount|yearmonth|amount 12M|
+-------------+-----------+----------+------+---------+----------+
|4421-222-222 |DE |2019-02-05|100 |230 |100 |
|4421-222-222 |DE |2019-03-01|50 |231 |150 |
|1234-567-890 |EN |2019-05-23|42 |233 |42 |
|1345-457-456 |EN |2019-12-12|107 |240 |107 |
|4421-222-222 |DE |2020-02-05|200 |242 |250 |
+-------------+-----------+----------+------+---------+----------+
我不确定确切的12个月,但这会起作用。
import pyspark.sql.functions as f
from pyspark.sql import Window
w = Window.partitionBy('articlenumber', 'countrycode').orderBy('unix_date').rangeBetween(- 365 * 86400, 0)
df.withColumn('unix_date', f.unix_timestamp('date', 'yyyy-MM-dd')) \
.withColumn('amount 12M', f.sum('amount').over(w)) \
.orderBy('date').show(10, False)
+-------------+-----------+----------+------+----------+----------+
|articlenumber|countrycode|date |amount|unix_date |amount 12M|
+-------------+-----------+----------+------+----------+----------+
|4421-222-222 |DE |2019-02-05|100 |1549324800|100 |
|4421-222-222 |DE |2019-02-06|50 |1549411200|150 |
|1234-567-890 |EN |2019-05-23|42 |1558569600|42 |
|1345-457-456 |EN |2019-12-12|107 |1576108800|107 |
|4421-222-222 |DE |2020-02-05|200 |1580860800|350 |
+-------------+-----------+----------+------+----------+----------+