становить точность отображения в PySpark при вызове.show()?

Рассмотрим следующий пример:

from math import sqrt
import pyspark.sql.functions as f

data = zip(
    map(lambda x: sqrt(x), range(100, 105)),
    map(lambda x: sqrt(x), range(200, 205))
)
df = sqlCtx.createDataFrame(data, ["col1", "col2"])
df.select([f.avg(c).alias(c) for c in df.columns]).show()

Какие выводы:

#+------------------+------------------+
#|              col1|              col2|
#+------------------+------------------+
#|10.099262230352151|14.212583322380274|
#+------------------+------------------+

Как я могу изменить его, чтобы он отображал только 3 цифры после десятичной точки?

Желаемый вывод:

#+------+------+
#|  col1|  col2|
#+------+------+
#|10.099|14.213|
#+------+------+

Это версия PySparkэтот вопрос скала, Я публикую его здесь, потому что я не смог найти ответ при поиске решений PySpark, и я думаю, что он может быть полезен другим в будущем.