如何从概率模型得到第一个元素得到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"变量。我需要此列中的第一个元素并将其粘贴到数据框中。
感谢您的帮助。