栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

Apache Spark:如何从DataFrame创建矩阵?

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Apache Spark:如何从DataFrame创建矩阵?

由于您没有提供示例输入,因此我假设它看起来或多或少像这样,其中

id
有一个行号并
image
包含值。

traindf = sqlContext.createDataframe([    (1, [1, 2, 3]),    (2, [4, 5, 6]),    (3, (7, 8, 9))], ("id", "image"))

您必须了解的第一件事

DenseMatrix
本地 数据结构。确切地说,它是一个包装
numpy.ndarray
。到目前为止(Spark
1.4.1)在PySpark MLlib中没有分布式等效项。

密集矩阵采取三个强制性参数

numRows
numCols
values
这里
values
是一个本地数据结构。在您的情况下,您必须先收集:

values = (traindf.    rdd.    map(lambda r: (r.id, r.image)). # Extract row id and data    sortByKey(). # Sort by row id    flatMap(lambda (id, image): image).    collect())ncol = len(traindf.rdd.map(lambda r: r.image).first())nrow = traindf.count()dm = DenseMatrix(nrow, ncol, values)

最后:

> print dm.toArray()[[ 1.  4.  7.] [ 2.  5.  8.] [ 3.  6.  9.]]

编辑

在Spark 1.5+中,您可以使用

mllib.linalg.distributed
以下方法:

from pyspark.mllib.linalg.distributed import IndexedRow, IndexedRowMatrixmat = IndexedRowMatrix(traindf.map(lambda row: IndexedRow(*row)))mat.numRows()## 4mat.numCols()## 3

尽管就目前而言,API仍然仅限于在实践中有用。



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/634311.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号