如何在pyspark中显示块矩阵乘法的结果?

cs7cruho  于 2021-07-12  发布在  Spark
关注(0)|答案(1)|浏览(525)

这听起来像是一个简单的问题,但我不知道如何将pyspark blockmatrix的内容显示到控制台。我应该调用什么方法来实际查看结果?

t2a7ltrp

t2a7ltrp1#

你可以打电话 toLocalMatrix() . 我使用spark mllib python api文档中的示例矩阵来说明:

mat = mat1.toLocalMatrix()

# which returns a DenseMatrix

# DenseMatrix(6, 2, [1.0, 2.0, 3.0, 7.0, 8.0, 9.0, 4.0, 5.0, 6.0, 10.0, 11.0, 12.0], 0)

# and could be further converted to a numpy array using `.toArray()`:

np_mat = mat.toArray()

# array([[ 1.,  4.],

# [ 2.,  5.],

# [ 3.,  6.],

# [ 7., 10.],

# [ 8., 11.],

# [ 9., 12.]])

相关问题