round(col, scale=0)
Описание
Функция round()
округляет число до указанного количества знаков после запятой.
Параметры
col
: Column - числовой столбец для округленияscale
: Integer - количество знаков после запятой (по умолчанию 0)
Возвращаемое значение
Double - округленное число
Пример использования
from pyspark.sql.functions import round
from pyspark.sql import SparkSession
# Создаем SparkSession
spark = SparkSession.builder.appName("round_example").getOrCreate()
# Создаем DataFrame
data = [
("A", 3.14159),
("B", 2.71828),
("C", 1.41421)
]
df = spark.createDataFrame(data, ["id", "value"])
# Округляем числа
result = df.select(
"id",
round("value").alias("round0"),
round("value", 1).alias("round1"),
round("value", 2).alias("round2")
).show()
# Результат:
# +---+------+------+------+
# |id |round0|round1|round2|
# +---+------+------+------+
# |A |3.0 |3.1 |3.14 |
# |B |3.0 |2.7 |2.72 |
# |C |1.0 |1.4 |1.41 |
# +---+------+------+------+