如何将两列pyspark数据帧相减并相除？

from pyspark.sql.functions import col

df = df.withColumn("diff", col("col1") - col("col2"))

df = df.withColumn("ratio", col("col1") / col("col2"))

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

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

# 创建示例数据帧
data = [("A", 10, 2), ("B", 5, 0), ("C", 8, 4)]
df = spark.createDataFrame(data, ["id", "col1", "col2"])

# 计算差和商
df = df.withColumn("diff", col("col1") - col("col2"))
df = df.withColumn("ratio", col("col1") / col("col2"))

# 显示结果
df.show()

+---+----+----+----+-----+
| id|col1|col2|diff|ratio|
+---+----+----+----+-----+
|  A|  10|   2|   8|  5.0|
|  B|   5|   0|   5| null|
|  C|   8|   4|   4|  2.0|
+---+----+----+----+-----+