如何在 PySpark window() 中使用毫秒作为参数。 function 之间的范围？

Question

我想在 PySpark 中的 Window() 中使用 function 之间的范围，并以毫秒为参数。

我正在尝试这样做：

df = df.withColumn("timestamp_ms", F.col("Dates").cast("double"))


w = (Window().orderBy("timestamp_ms").rangeBetween(-0.1, 0.0))

但是，由于 rangeBetween 方法中的浮点参数（-0.1），我得到了一个错误。

py4j.protocol.Py4JError: An error occurred while calling o73.rangeBetween. Trace:
py4j.Py4JException: Method rangeBetween([class java.lang.Double, class java.lang.Double]) does not exist
        at py4j.reflection.ReflectionEngine.getMethod(ReflectionEngine.java:318)
        at py4j.reflection.ReflectionEngine.getMethod(ReflectionEngine.java:326)
        at py4j.Gateway.invoke(Gateway.java:274)
        at py4j.commands.AbstractCommand.invokeMethod(AbstractCommand.java:132)
        at py4j.commands.CallCommand.execute(CallCommand.java:79)
        at py4j.GatewayConnection.run(GatewayConnection.java:238)
        at java.lang.Thread.run(Thread.java:748)

有没有其他方法可以在（-0.1 秒，0）之间创建这些 windows？

*timestamp_ms 是一列具有毫秒分辨率的时间戳

*Dates 是具有以下格式的日期的列：“2019-07-26 08:56:07.171”

谢谢！

Answer 1

我这样做解决了这个问题：

df = df.withColumn("timestamp_ms", F.col("Dates").cast("double"))

df = df.withColumn("timestamp_ms", F.col("timestamp_ms")*1000)

w = (Window().orderBy("timestamp_ms").rangeBetween(-100, 0))

我注意到，如果我将“timestamp_ms”列乘以 1000，我将在 integer dtype 中得到一列时间戳，分辨率为毫秒。 现在我可以使用 rangeBetween (-0.1 secs, 0) 来创建 Window()。