python - 从数据框中获取值(value)

Question

在 Scala 中，我可以执行 get(#) 或 getAs[Type](#) 从数据帧中获取值。我应该如何在 pyspark 中执行此操作？

我有两列 DataFrame:item(string) 和 salesNum(integers)。我做了一个 groupby 和 mean 来得到这些数字的平均值，如下所示:

saleDF.groupBy("salesNum").mean()).collect()

并且有效。现在，我在具有一个值的数据框中得到了平均值。

我如何从数据框中获取该值以获取 float 的平均值？

Answer 1

collect() 将您的结果作为 python 列表返回。要从列表中获取值，您只需要像这样获取第一个元素:

saleDF.groupBy("salesNum").mean()).collect()[0]