如何从概率中获取元素在pyspark模型中进行列预测？
如何从概率中获取元素在pyspark模型中进行列预测？
提问于 2018-06-05 11:05:46
怎样才能从概率模型的第一个元素得到一个pyspark数据框？
+------+--------------------+
|labelh| probability|
+------+--------------------+
| 1|[0.72498853530094...|
| 1|[0.99989771872286...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
| 1|[0.72498853530094...|
+------+--------------------+
概率列有两个元素，是类型“vector”变量。我需要此列中的第一个元素并将其粘贴到数据框中。
回答 1
Stack Overflow用户
发布于 2018-06-05 20:42:40
可以使用getItem()通过索引从矢量中提取值。在这种情况下，需要索引0处的项目：
from pyspark.sql.functions import col
df.select(col("probability").getItem(0))
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/-100004755
复制相关文章
相似问题