apache-spark - 舍入数据框中的所有列 - 小数点后两位 pyspark

Question

我有这个命令让我的数据框中的所有列四舍五入到小数点后两位:

data = data.withColumn("columnName1", func.round(data["columnName1"], 2))

我不知道如何通过一个命令(不是每一列分开)舍入所有数据框。有人可以帮帮我吗？我不想使用不同的列名称使用相同的命令 50 次。

Answer 1

没有将所有函数应用于列的函数或命令，但您可以迭代。

+-----+-----+
| col1| col2|
+-----+-----+
|1.111|2.222|
+-----+-----+

df = spark.read.option("header","true").option("inferSchema","true").csv("test.csv")

for c in df.columns:
    df = df.withColumn(c, round(c, 2))
    
df.show()

+----+----+
|col1|col2|
+----+----+
|1.11|2.22|
+----+----+