在数组python/pyspark中创建两个相邻数字之间的col值差异数组

def get_differences(arr):
    differences = []
    for i in range(len(arr)-1):
        diff = arr[i+1] - arr[i]
        differences.append(diff)
    return differences

# 示例用法
arr = [1, 3, 5, 7, 9]
result = get_differences(arr)
print(result)

from pyspark.sql import SparkSession
from pyspark.sql.functions import lag, col

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 创建DataFrame
data = [(1,), (3,), (5,), (7,), (9,)]
df = spark.createDataFrame(data, ["col"])

# 添加差异列
df = df.withColumn("diff", col("col") - lag("col").over(Window.orderBy("col")))

# 显示结果
df.show()