将Spark DataFrame转换为float

Question

我有一些DataFrame：

data=[('Alice',1),('Bob',2),('Alice',5)]

并计算一个像这样的平均值：

value = dataF.groupBy().mean().collect()

结果：

[Row(avg(age)=2.6666666666666665)]

如何将其转换为浮动？当我使用round函数时：

print round(value,2)

我收到TypeError: a float is required错误。

Answer 1

value = dataF.groupBy（）。mean（）。head（）[0]

Answer 2

对于简单转换，可以执行类似

的操作

val test_df = dataDF.withColumn("xx", $"yy".cast("float"))

如果需要就地转换，也可以使用UDF。希望这会有所帮助。

Answer 3

value =（round（float（dataF.groupBy（）。avg（'age'）。first（）[0]），2））