如何在Spark DataFrame中强制将双精度值设置为特定位数？

from pyspark.sql import SparkSession
from pyspark.sql.functions import format_number

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

# 创建示例DataFrame
data = [(1, 3.14159), (2, 2.71828), (3, 1.41421)]
df = spark.createDataFrame(data, ["id", "value"])

# 使用format_number函数将value列保留两位小数
df = df.withColumn("formatted_value", format_number("value", 2))

# 显示结果
df.show()

+---+--------+---------------+
| id|   value|formatted_value|
+---+--------+---------------+
|  1|3.14159|           3.14|
|  2|2.71828|           2.72|
|  3|1.41421|           1.41|
+---+--------+---------------+