python之是否有任何 pyspark 函数可以添加下个月，如 DATE_ADD(date, month(int type))

我是 spark 的新手，是否有任何内置函数可以显示当前日期的下个月日期，比如今天是 27-12-2016，那么该函数将返回 27-01-2017。我使用了 date_add() 但没有添加月份的功能。我试过 date_add(date, 31) 但如果这个月有 30 天怎么办。

spark.sql("select date_add(current_date(),31)") .show() 

谁能帮我解决这个问题。我需要为此编写自定义函数吗？因为我仍然没有找到任何内置代码 提前致谢 卡莲

请您参考如下方法:

我找到的最直接的数据框友好解决方案，用于添加/减去月份

from pyspark.sql import functions as F 
# assume df has "current_date" column as type DateType 
months_to_add = 1  # int value, could be negative 
df = df.withColumn("new_date", F.add_months("current_date", months_to_add)) 

此结果将包括之前包含在 df 中的任何其他列。