DenseVector到数组列表Pyspark ML

Question

我有一个变量dense_vector，当我打印它时会打印出DenseVector([0.0, 0.0, 0.9998])

有没有办法只获取数组[0.0, 0.0, 0.9998]？

Answer 1

在您的情况下 - 至少根据我发现的the docs - 您应该能够使用DenseVector.values访问这些值。

一般答案：

对象的字符串表示形式不得无形地反映对象的内容。数据可以存储为类实例中的列表，元组，字典，属性......，因此最好的办法是查看__str__ / __repr__实现并自行构建列表。

或者，如果它是您自己的类，则创建一个返回列表的方法，或者您甚至可以对该类进行子类化，然后在您自己的类上添加一个返回列表的方法。