круглые результаты в сводных таблицах результатов (pyspark)

Question

Здравствуйте, как бы я округлил это содержимое таблицы, выведенной этим кодом.

from pyspark.sql.functions import *
exprs = {x: "sum" for x in data2.columns[:4]}
data2.groupBy("Species").agg(exprs).show() 

Я пробовал

round(data2.groupBy("Species").agg(exprs),2).show() #not ok

data2.groupBy("Species").agg(exprs).show().round(2) # not ok

Answer 1

раунд работает только на один столбец. Поэтому вы должны вызывать его для каждого столбца, например,

agg_cols = data2.columns[:4]
exprs = [sum(col(x)).alias(x) for x in agg_cols]
aggregated_df = data2.groupBy("Species").agg(*exprs)
aggregated_df.select(col("Species"), *[round(c, 2) for c in agg_cols]).show()