在Pyspark中对数据框进行舍入

from pyspark.sql import SparkSession
from pyspark.sql.functions import round

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

# 创建示例数据框
data = [("Alice", 3.14159), ("Bob", 2.71828), ("Charlie", 1.41421)]
df = spark.createDataFrame(data, ["Name", "Value"])

# 对数据框中的Value列进行舍入，保留两位小数
df_rounded = df.withColumn("RoundedValue", round(df["Value"], 2))

# 显示结果
df_rounded.show()

+-------+-------+------------+
|   Name|  Value|RoundedValue|
+-------+-------+------------+
|  Alice|3.14159|        3.14|
|    Bob|2.71828|        2.72|
|Charlie|1.41421|        1.41|
+-------+-------+------------+