如何在Spark RDD(Java)中按索引获取元素

progNewbie:

我知道方法rdd.first()会给我RDD中的第一个元素。

还有rdd.take(num)方法,它给了我第一个“ num”元素。

但是,是否有可能通过索引获取元素?

谢谢。

网格:

首先应为RDD建立索引,这应该是可能的。转换zipWithIndex提供了稳定的索引,以其原始顺序对每个元素进行编号。

鉴于: rdd = (a,b,c)

val withIndex = rdd.zipWithIndex // ((a,0),(b,1),(c,2))

要通过索引查找元素,此形式无用。首先,我们需要使用索引作为键:

val indexKey = withIndex.map{case (k,v) => (v,k)}  //((0,a),(1,b),(2,c))

现在,可以使用lookupPairRDD中操作按键查找元素:

val b = indexKey.lookup(1) // Array(b)

如果您希望lookup经常在同一RDD上使用,建议您缓存indexKeyRDD以提高性能。

如何使用Java API进行此操作是读者的一项练习。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章