您将需要使用几个窗口函数。在3个不同的窗口中计数标志：-2:0，-1:1，0:-2。如果这些中至少有一个的sum是3，那么你有3个连续的1。
在下面的脚本中，我假设日期以 string 数据类型存储，因此我使用了列表达式date从字符串中读取真实日期。如果日期不是字符串格式，则不应使用该行。
输入：

from pyspark.sql import functions as F, Window as W
df = spark.createDataFrame(
    [('01-01-2022', 1),
     ('02-01-2022', 1),
     ('03-01-2022', 1),
     ('04-01-2022', 1),
     ('05-01-2022', 0),
     ('06-01-2022', 0),
     ('07-01-2022', 1),
     ('08-01-2022', 1),
     ('09-01-2022', 0),
     ('10-01-2022', 1)],
    ['date', 'flag'])

脚本：

date = F.to_date('date', 'dd-MM-yyyy')
cond1 = F.sum('flag').over(W.orderBy(date).rowsBetween(-2, 0)) == 3
cond2 = F.sum('flag').over(W.orderBy(date).rowsBetween(-1, 1)) == 3
cond3 = F.sum('flag').over(W.orderBy(date).rowsBetween(0, 2)) == 3
df = df.withColumn('flag', (cond1 | cond2 | cond3).cast('long'))

df.show()

# +----------+----+

# |      date|flag|

# +----------+----+

# |01-01-2022|   1|

# |02-01-2022|   1|

# |03-01-2022|   1|

# |04-01-2022|   1|

# |05-01-2022|   0|

# |06-01-2022|   0|

# |07-01-2022|   0|

# |08-01-2022|   0|

# |09-01-2022|   0|

# |10-01-2022|   0|

# +----------+----+