Spark核心类：弹性分布式数据集RDD及其转换和操作pyspark.RDD

Myth丶恋晨 2022-07-14 00:17 167阅读 0赞

http://[blog.csdn.net/pipisorry/article/details/53257188][blog.csdn.net_pipisorry_article_details_53257188]

## 弹性分布式数据集RDD(Resilient Distributed Dataset) ##

## 术语定义 ##

l弹性分布式数据集（RDD）： Resillient Distributed Dataset，Spark的基本计算单元，可以通过一系列算子进行操作（主要有Transformation和Action操作）；

l有向无环图（DAG）：Directed Acycle graph，反应RDD之间的依赖关系；

l有向无环图调度器（DAG Scheduler）：根据Job构建基于Stage的DAG，并提交Stage给TaskScheduler；

l任务调度器（Task Scheduler）：将Taskset提交给worker（集群）运行并回报结果；

l窄依赖（Narrow dependency）：子RDD依赖于父RDD中固定的data partition；

l宽依赖（Wide Dependency）：子RDD对父RDD中的所有data partition都有依赖。

## RDD概念 ##

RDD是Spark的最基本抽象,是对分布式内存的抽象使用，实现了以操作本地集合的方式来操作分布式数据集的抽象实现。RDD是Spark最核心的东西，它表示已被分区，不可变的并能够被并行操作的数据集合，不同的数据集格式对应不同的RDD实现。RDD必须是可序列化的。RDD可以cache到内存中，每次对RDD数据集的操作之后的结果，都可以存放到内存中，下一个操作可以直接从内存中输入，省去了MapReduce大量的磁盘IO操作。这对于迭代运算比较常见的机器学习算法, 交互式数据挖掘来说，效率提升非常大。

RDD 最适合那种在数据集上的所有元素都执行相同操作的批处理式应用。在这种情况下， RDD 只需记录血统中每个转换就能还原丢失的数据分区，而无需记录大量的数据操作日志。所以 RDD 不适合那些需要异步、细粒度更新状态的应用 ，比如 Web 应用的存储系统，或增量式的 Web 爬虫等。对于这些应用，使用具有事务更新日志和数据检查点的数据库系统更为高效。

### RDD的特点 ###

1.来源：一种是从持久存储获取数据，另一种是从其他RDD生成

2.只读：状态不可变，不能修改

3.分区：支持元素根据 Key 来分区 ( Partitioning ) ，保存到多个结点上，还原时只会重新计算丢失分区的数据，而不会影响整个系统

4.路径：在 RDD 中叫世族或血统 ( lineage ) ，即 RDD 有充足的信息关于它是如何从其他 RDD 产生而来的

5.持久化：可以控制存储级别（内存、磁盘等）来进行持久化

6.操作：丰富的动作 ( Action ) ，如Count、Reduce、Collect和Save 等

### RDD基础数据类型 ###

目前有两种类型的基础RDD：并行集合（Parallelized Collections）：接收一个已经存在的Scala集合，然后进行各种并行计算。 Hadoop数据集（Hadoop Datasets）：在一个文件的每条记录上运行函数。只要文件系统是HDFS，或者hadoop支持的任意存储系统即可。这两种类型的RDD都可以通过相同的方式进行操作，从而获得子RDD等一系列拓展，形成lineage血统关系图。

**1.** **并行化集合**

并行化集合是通过调用SparkContext的parallelize方法，在一个已经存在的Scala集合上创建的（一个Seq对象）。集合的对象将会被拷贝，创建出一个可以被并行操作的分布式数据集。例如，下面的解释器输出，演示了如何从一个数组创建一个并行集合。

例如：val rdd = sc.parallelize(Array(1 to 10)) 根据能启动的executor的数量来进行切分多个slice，每一个slice启动一个Task来进行处理。

val rdd = sc.parallelize(Array(1 to 10), 5) 指定了partition的数量

**2.** **Hadoop数据集**

Spark可以将任何Hadoop所支持的存储资源转化成RDD,如本地文件（需要网络文件系统，所有的节点都必须能访问到）、HDFS、Cassandra、HBase、Amazon S3等，Spark支持文本文件、SequenceFiles和任何Hadoop InputFormat格式。

（1）使用textFile()方法可以将本地文件或HDFS文件转换成RDD

支持整个文件目录读取，文件可以是文本或者压缩文件(如gzip等，自动执行解压缩并加载数据)。如textFile（”file:///dfs/data”）

支持通配符读取,例如：

*val rdd1 = sc.textFile("file:///root/access\_log/access\_log\*.filter");*

*val rdd2=rdd1.map(\_.split("t")).filter(\_.length==6)*

*rdd2.count()*

*......*

*14/08/20 14:44:48 INFO HadoopRDD: Input split: file:/root/access\_log/access\_log.20080611.decode.filter:134217728+20705903*

*......*

textFile()可选第二个参数slice，默认情况下为每一个block分配一个slice。用户也可以通过slice指定更多的分片，但不能使用少于HDFS block的分片数。

（2）使用wholeTextFiles()读取目录里面的小文件，返回（用户名、内容）对

（3）使用sequenceFile\[K,V\]()方法可以将SequenceFile转换成RDD。SequenceFile文件是Hadoop用来存储二进制形式的key-value对而设计的一种平面文件(Flat File)。

（4）使用SparkContext.hadoopRDD方法可以将其他任何Hadoop输入类型转化成RDD使用方法。一般来说，HadoopRDD中每一个HDFS block都成为一个RDD分区。

此外，通过Transformation可以将HadoopRDD等转换成FilterRDD(依赖一个父RDD产生）和JoinedRDD（依赖所有父RDD）等。

[皮皮blog][blog]

# Rdd的创建和操作    #

## 创建RDD ##

### 并行集合 (Parallelized collections)    ###

RDD可从现有的集合创建。比如在Scala shell中:  
val collection = List("a", "b", "c", "d", "e")  
val rddFromCollection = sc.parallelize(collection)

并行集合 (Parallelized collections) 的创建是通过在一个已有的集合(Scala Seq)上调用 SparkContext 的 parallelize 方法实现的。集合中的元素被复制到一个可并行操作的分布式数据集中。并行集合一个很重要的参数是切片数(slices)，表示一个数据集切分的份数。Spark 会在集群上为每一个切片运行一个任务。你可以在集群上为每个 CPU 设置 2-4 个切片(slices)。正常情况下，Spark 会试着基于你的集群状况自动地设置切片的数目。然而，你也可以通过 parallelize 的第二个参数手动地设置(例如：sc.parallelize(data, 10))。

### 外部数据集    ###

比如Spark 可以从任何一个 Hadoop 支持的存储源创建分布式数据集，包括你的本地文件系统，HDFS，Cassandra，HBase，[Amazon S3][]等。 Spark 支持文本文件(text files)，[SequenceFiles][] 和其他 Hadoop [InputFormat][]。

用一个本地文件系统里的文件创建RDD:  
val rddFromTextFile = sc.textFile("LICENSE")

\[[外部数据集][Link 1]\]

序列化及其反序列化

saveAsPickleFile(path, batchSize=10)

Save this RDD as a SequenceFile of serialized objects. The serializerused is pyspark.serializers.PickleSerializer, default batch sizeis 10.

>>> tmpFile = NamedTemporaryFile(delete=True)
    >>> tmpFile.close()
    >>> sc.parallelize([1, 2, 'spark', 'rdd']).saveAsPickleFile(tmpFile.name, 3)
    >>> sorted(sc.pickleFile(tmpFile.name, 5).map(str).collect())
    ['1', '2', 'rdd', 'spark']

\[pickleFile(name, minPartitions=None) [¶][Link 2]\]

## Spark操作    ##

在Spark编程模式下,所有的操作被分为转换(transformation)和执行(action)两种。

一般来说,转换操作是对一个数据集里的所有记录执行某种函数,从而使记录发生改变;而执行通常是运行某些计算或聚合操作,并将结果返回运行 SparkContext 的那个驱动程序。

Spark中的转换操作是延后的，也就是说,在RDD上调用一个转换操作并不会立即触发相应的计算。相反,这些转换操作会链接起来,并只在有执行操作被调用时才被高效地计算。这样,大部分操作可以在集群上并行执行,只有必要时才计算结果并将其返回给驱动程序（如执行操作count）,从而提高了Spark的效率。

例如，map 是一个转换操作，它将每一个数据集元素传递给一个函数并且返回一个新的 RDD。另一方面，reduce 是一个动作，它使用相同的函数来聚合 RDD 的所有元素，并且将最终的结果返回到驱动程序(不过也有一个并行 reduceByKey 能返回一个分布式数据集)。例如，我们可以实现：通过 map 创建一个新数据集在 reduce 中使用，并且仅仅返回 reduce 的结果给 driver，而不是整个大的映射过的数据集。

### 转换操作    ###

<table> 
 <tbody> 
 <tr> 
 <td>map(func)</td> 
 <td>返回一个新的分布式数据集，由每个原元素经过func函数转换后组成</td> 
 </tr> 
 <tr> 
 <td>filter(func)</td> 
 <td>返回一个新的数据集，由经过func函数后返回值为true的原元素组成</td> 
 </tr> 
 <tr> 
 <td>flatMap(func)</td> 
 <td>类似于map，但是每一个输入元素，会被映射为0到多个输出元素（因此，func函数的返回值是一个Seq，而不是单一元素）</td> 
 </tr> 
 <tr> 
 <td>sample(withReplacement,&nbsp; frac, seed)</td> 
 <td>根据给定的随机种子seed，随机抽样出数量为frac的数据</td> 
 </tr> 
 <tr> 
 <td>union(otherDataset)</td> 
 <td>返回一个新的数据集，由原数据集和参数联合而成</td> 
 </tr> 
 <tr> 
 <td>groupByKey([numTasks])</td> 
 <td>在一个由（K,V）对组成的数据集上调用，返回一个（K，Seq[V])对的数据集。注意：默认情况下，使用8个并行任务进行分组，你可以传入numTask可选参数，根据数据量设置不同数目的Task</td> 
 </tr> 
 <tr> 
 <td>reduceByKey(func,&nbsp; [numTasks])</td> 
 <td>在一个（K，V)对的数据集上使用，返回一个（K，V）对的数据集，key相同的值，都被使用指定的reduce函数聚合到一起。和groupbykey类似，任务的个数是可以通过第二个可选参数来配置的。</td> 
 </tr> 
 <tr> 
 <td>join(otherDataset,&nbsp; [numTasks])</td> 
 <td>在类型为（K,V)和（K,W)类型的数据集上调用，返回一个（K,(V,W))对，每个key中的所有元素都在一起的数据集</td> 
 </tr> 
 <tr> 
 <td>groupWith(otherDataset,&nbsp; [numTasks])</td> 
 <td>在类型为（K,V)和(K,W)类型的数据集上调用，返回一个数据集，组成元素为（K, Seq[V], Seq[W]) Tuples。这个操作在其它框架，称为CoGroup</td> 
 </tr> 
 <tr> 
 <td>cartesian(otherDataset)</td> 
 <td>笛卡尔积。但在数据集T和U上调用时，返回一个(T，U）对的数据集，所有元素交互进行笛卡尔积。</td> 
 </tr> 
 </tbody> 
</table>

### **执行操作**    ###

<table> 
 <tbody> 
 <tr> 
 <td>reduce(func) </td> 
 <td>通过函数func聚集数据集中的所有元素。Func函数接受2个参数，返回一个值。这个函数必须是关联性的，确保可以被正确的并发执行</td> 
 </tr> 
 <tr> 
 <td>collect()</td> 
 <td>在Driver的程序中，以数组的形式，返回数据集的所有元素。这通常会在使用filter或者其它操作后，返回一个足够小的数据子集再使用，直接将整个RDD集Collect返回，很可能会让Driver程序OOM。返回所有元素到驱动程序</td> 
 </tr> 
 <tr> 
 <td>count()</td> 
 <td>返回数据集的元素个数</td> 
 </tr> 
 <tr> 
 <td>take(n)</td> 
 <td>返回一个数组，由数据集的前n个元素组成。注意，这个操作目前并非在多个节点上，并行执行，而是Driver程序所在机器，单机计算所有的元素(Gateway的内存压力会增大，需要谨慎使用）。返回前k个元素到驱动程序</td> 
 </tr> 
 <tr> 
 <td>first()</td> 
 <td>返回数据集的第一个元素（类似于take（1）；也就是返回第1个元素到驱动程序</td> 
 </tr> 
 <tr> 
 <td>saveAsTextFile(path)</td> 
 <td>将数据集的元素，以textfile的形式，保存到本地文件系统，hdfs或者任何其它hadoop支持的文件系统。Spark将会调用每个元素的toString方法，并将它转换为文件中的一行文本</td> 
 </tr> 
 <tr> 
 <td>saveAsSequenceFile(path)</td> 
 <td>将数据集的元素，以sequencefile的格式，保存到指定的目录下，本地系统，hdfs或者任何其它hadoop支持的文件系统。RDD的元素必须由key-value对组成，并都实现了Hadoop的Writable接口，或隐式可以转换为Writable（Spark包括了基本类型的转换，例如Int，Double，String等等）</td> 
 </tr> 
 <tr> 
 <td>foreach(func)</td> 
 <td>在数据集的每一个元素上，运行函数func。这通常用于更新一个累加器变量，或者和外部存储系统做交互</td> 
 </tr> 
 </tbody> 
</table>

**注意事项**

通常只在需将结果返回到驱动程序所在节点以供本地处理时,才调用 collect 函数。

注意, collect 函数一般仅在的确需要将整个结果集返回驱动程序并进行后续处理时才有必要调用。如果在一个非常大的数据集上调用该函数,可能耗尽驱动程序的可用内存,进而导致程序崩溃。  
高负荷的处理应尽可能地在整个集群上进行,从而避免驱动程序成为系统瓶颈。然而在不少情况下,将结果收集到驱动程序的确是有必要的。很多机器学习算法的迭代过程便属于这类情况。

\[RDD支持的转换和执行操作的完整列表以及更为详细的例子

参见《Spark编程指南[Spark Programming Guide][]》

以及Spark API([Scala][])文档\]

### RDD缓存策略 ###

默认情况下，每一个转换过的 RDD 会在每次执行动作(action)的时候重新计算一次。

然而，你能通过persist()或者cache()方法持久化一个rdd。首先，在action中计算得到rdd；然后，将其保存在每个节点的内存中。在这个情况下，Spark 会在集群上保存相关的元素，在你下次查询的时候会变得更快。在这里也同样支持持久化 RDD 到磁盘，或在多个节点间复制。

Spark的缓存是一个容错的技术-如果RDD的任何一个分区丢失，它可以通过原有的转换（transformations）操作自动的重复计算并且创建出这个分区。

rddFromTextFile.cache

调用一个RDD的 cache 函数将会告诉Spark将这个RDD缓存在内存中。在RDD首次调用一个执行操作时,这个操作对应的计算会立即执行,数据会从数据源里读出并保存到内存。因此,首次调用 cache 函数所需要的时间会部分取决于Spark从输入源读取数据所需要的时间。但是,当下一次访问该数据集的时候,数据可以直接从内存中读出从而减少低效的I/O操作,加快计算。多数情况下,这会取得数倍的速度提升。

val lineLengths = lines.map(s => s.length)
    val totalLength = lineLengths.reduce((a, b) => a + b)

如果我们想要再次使用 `lineLengths`，我们可以添加：

lineLengths.persist()

我们可以利用不同的存储级别存储每一个被持久化的RDD。例如，它允许我们持久化集合到磁盘上、将集合作为序列化的Java对象持久化到内存中、在节点间复制集合或者存储集合到Tachyon中。我们可以通过传递一个StorageLevel对象给persist()方法设置这些存储级别。cache()方法使用了默认的存储级别—StorageLevel.MEMORY\_ONLY。

\[[RDD 持久化][RDD]\]

### 删除数据 ###

Spark自动的监控每个节点缓存的使用情况，利用最近最少使用原则删除老旧的数据。如果你想手动的删除RDD，可以使用RDD.unpersist()方法 [  
][blog]

[皮皮blog][blog]

# 转换函数详解 #

### map(f, preservesPartitioning=False) ###

Return a new RDD by applying a function to each element of this RDD.

>>> rdd = sc.parallelize(["b", "a", "c"])
    >>> sorted(rdd.map(lambda x: (x, 1)).collect())
    [('a', 1), ('b', 1), ('c', 1)]

map接收的函数最好是

def mapper(value):
        ...
        return key, value

map 函数,我们将每一个字符串都转换为一个整数,从而返回一个由若干 Int 构成的RDD对象。  
val intsFromStringsRDD = rddFromTextFile.map(line => line.size)

### flatMap(f, preservesPartitioning=False) ###

Return a new RDD by first applying a function to all elements of thisRDD, and then flattening the results.

>>> rdd = sc.parallelize([2, 3, 4])
    >>> sorted(rdd.flatMap(lambda x: range(1, x)).collect())
    [1, 1, 1, 2, 2, 3]
    >>> sorted(rdd.flatMap(lambda x: [(x, x), (x, x)]).collect())
    [(2, 2), (2, 2), (3, 3), (3, 3), (4, 4), (4, 4)]

Note: use flatMap if you want a map function that returns multiple outputs.The function passed to flatMap can return an iterable.

这样的话map函数可以返回一个可迭代对象，相当于将返回值变成了多个rdd行（本来返回一行rdd，但是却是多个(k, v)组合而不是一个可供reducebykey使用的单个(k, v)）？

def mapper(value):
        ...
        result_list = []
        for key, value in some_list:
            result_list.append( key, value )
        return result_list

\[[List (or iterator) of tuples returned by MAP (PySpark)][List _or iterator_ of tuples returned by MAP _PySpark]\]

### mapValues(f) ###

Pass each value in the key-value pair RDD through a map functionwithout changing the keys; this also retains the original RDD’spartitioning.

>>> x = sc.parallelize([("a", ["apple", "banana", "lemon"]), ("b", ["grapes"])])
    >>> def f(x): return len(x)
    >>> x.mapValues(f).collect()
    [('a', 3), ('b', 1)]

### combineByKey(createCombiner, mergeValue, mergeCombiners, numPartitions=None, partitionFunc=<function portable\_hash>) ###

Generic function to combine the elements for each key using a customset of aggregation functions.

Turns an RDD\[(K, V)\] into a result of type RDD\[(K, C)\], for a “combinedtype” C. Note that V and C can be different – for example, one mightgroup an RDD of type (Int, Int) into an RDD of type (Int, List\[Int\]).

Users provide three functions:

*  createCombiner, which turns a V into a C (e.g., createsa one-element list)
 *  mergeValue, to merge a V into a C (e.g., adds it to the end ofa list)
 *  mergeCombiners, to combine two C’s into a single one.

In addition, users can control the partitioning of the output RDD.

>>> x = sc.parallelize([("a", 1), ("b", 1), ("a", 1)])
    >>> def add(a, b): return a + str(b)
    >>> sorted(x.combineByKey(str, add, add).collect())
    [('a', '11'), ('b', '1')]

\[combineByKey(createCombiner, mergeValue, mergeCombiners, numPartitions=None, partitionFunc=<function portable\_hash at 0x7ff5681b9d70>)[¶][Link 3]\]

### reduce(f) ###

Reduces the elements of this RDD using the specified commutative andassociative binary operator. Currently reduces partitions locally.

>>> from operator import add
    >>> sc.parallelize([1, 2, 3, 4, 5]).reduce(add)
    15
    >>> sc.parallelize((2 for _ in range(10))).map(lambda x: 1).cache().reduce(add)
    10

### reduceByKey(func, numPartitions=None, partitionFunc=<function portable\_hash at 0x7ff5681b9d70>) ###

Merge the values for each key using an associative and commutative reduce function.

This will also perform the merging locally on each mapper before sending results to a reducer, similarly to a “combiner” in MapReduce.

Output will be partitioned with numPartitions partitions, orthe default parallelism level if numPartitions is not specified.Default partitioner is hash-partition.

>>> from operator import add
    >>> rdd = sc.parallelize([("a", 1), ("b", 1), ("a", 1)])
    >>> sorted(rdd.reduceByKey(add).collect())
    [('a', 2), ('b', 1)]

### filter(f) ###

Return a new RDD containing only the elements that satisfy a predicate.

>>> rdd = sc.parallelize([1, 2, 3, 4, 5])
    >>> rdd.filter(lambda x: x % 2 == 0).collect()
    [2, 4]

### collect() ###

Return a list that contains all of the elements in this RDD.Note that this method should only be used if the resulting array is expectedto be small, as all the data is loaded into the driver’s memory.

### collectAsMap() ###

Return the key-value pairs in this RDD to the master as a dictionary.

Note that this method should only be used if the resulting data is expectedto be small, as all the data is loaded into the driver’s memory.

>>> m = sc.parallelize([(1, 2), (3, 4)]).collectAsMap()
    >>> m[1]
    2

from: [http://blog.csdn.net/pipisorry/article/details/53257188][blog.csdn.net_pipisorry_article_details_53257188]

ref: \[class pyspark.RDD(jrdd, ctx, jrdd\_deserializer=AutoBatchedSerializer(PickleSerializer()))[¶][Link 4]\]

[blog.csdn.net_pipisorry_article_details_53257188]: http://blog.csdn.net/pipisorry/article/details/53257188
[blog]: http://blog.csdn.net/pipisorry
[Amazon S3]: http://wiki.apache.org/hadoop/AmazonS3
[SequenceFiles]: http://hadoop.apache.org/docs/current/api/org/apache/hadoop/mapred/SequenceFileInputFormat.html
[InputFormat]: http://hadoop.apache.org/docs/stable/api/org/apache/hadoop/mapred/InputFormat.html
[Link 1]: https://endymecy.gitbooks.io/spark-programming-guide-zh-cn/content/programming-guide/rdds/external-datasets.html
[Link 2]: http://spark.apache.org/docs/latest/api/python/pyspark.html#pyspark.SparkContext.pickleFile
[Spark Programming Guide]: http://spark.apache.org/docs/latest/programming-guide.html#rdd
[Scala]: http://spark.apache.org/docs/latest/api/scala/index.html#org.apache.spark.rdd.RDD
[RDD]: https://endymecy.gitbooks.io/spark-programming-guide-zh-cn/content/programming-guide/rdds/rdd-persistences.html
[List _or iterator_ of tuples returned by MAP _PySpark]: http://stackoverflow.com/questions/21096432/list-or-iterator-of-tuples-returned-by-map-pyspark
[Link 3]: http://spark.apache.org/docs/latest/api/python/pyspark.html#pyspark.RDD.combineByKey
[Link 4]: http://spark.apache.org/docs/latest/api/python/pyspark.html#pyspark.RDD