PySpark之RDD入门最全攻略！
众所周知，Spark的核心是RDD（Resilient Distributed Dataset）即弹性分布式数据集，属于一种分布式的内存系统的数据集应用。Spark主要优势就是来自RDD本身的特性，RDD能与其他系统兼容，可以导入外部存储系统的数据集，例如，HDFS、HBase或者其他Hadoop数据源。
1、RDD的基本运算
RDD运算类型 | 说明 |
|---|---|
转换（Transformation） | 转换运算将一个RDD转换为另一个RDD，但是由于RDD的lazy特性，转换运算不会立刻实际执行，它会等到执行到“动作”运算，才会实际执行。 |
动作（Action） | RDD执行动作运算之后，不会产生另一个RDD，它会产生数值、数组或写入文件系统；RDD执行动作运算后会立刻实际执行，并且连同之前的转换运算一起执行。 |
持久化（Persistence） | 对于那些会重复使用的RDD， 可以将RDD持久化在内存中作为后续使用，以提高执行性能。 |
2、基本RDD“转换”运算
首先我们要导入PySpark并初始化Spark的上下文环境：
初始化
from pyspark import SparkConf, SparkContext
sc = SparkContext()创建RDD
接下来我们使用parallelize方法创建一个RDD：
intRDD = sc.parallelize([3,1,2,5,5])stringRDD = sc.parallelize(['Apple','Orange','Grape','Banana','Apple'])RDD转换为Python数据类型
RDD类型的数据可以使用collect方法转换为python的数据类型：
print (intRDD.collect())
print (stringRDD.collect())输出为：
[3, 1, 2, 5, 5]
['APPLE', 'Orange', 'Grape', 'Banana','Apple']map运算
map运算可以通过传入的函数，将每一个元素经过函数运算产生另外一个RDD。 比如下面的代码中，将intRDD中的每个元素加1之后返回，并转换为python数组输出：
print (intRDD.map(lambda x:x+1).collect())结果为：
[4, 2, 3, 6, 6]filter运算
filter可以用于对RDD内每一个元素进行筛选，并产生另外一个RDD。 下面的例子中，我们筛选intRDD中数字小于3的元素，同事筛选stringRDD中包含ra的字符串：
print (intRDD.filter(lambda x: x<3).collect())
print (stringRDD.filter(lambda x:'ra' in x).collect())输出为：
[1, 2]
['Orange', 'Grape']distinct运算
distinct运算会删除重复的元素，比如我们去除intRDD中的重复元素1：
print (intRDD.distinct().collect())输出为：
[1, 2, 3, 5]randomSplit运算
randomSplit 运算将整个集合以随机数的方式按照比例分为多个RDD，比如按照0.4和0.6的比例将intRDD分为两个RDD，并输出：
sRDD = intRDD.randomSplit([0.4,0.6])print (len(sRDD))print (sRDD[0].collect())print (sRDD[1].collect())输出为：
2[3, 1]
[2, 5, 5]groupBy运算
groupBy运算可以按照传入匿名函数的规则，将数据分为多个Array。比如下面的代码将intRDD分为偶数和奇数：
result = intRDD.groupBy(lambda x : x % 2).collect()
print (sorted([(x, sorted(y)) for (x, y) in result]))输出为：
[(0, [2]), (1, [1, 3, 5, 5])]3、多个RDD转换运算
RDD也支持执行多个RDD的运算，这里，我们定义三个RDD：
intRDD1 = sc.parallelize([3,1,2,5,5])
intRDD2 = sc.parallelize([5,6])
intRDD3 = sc.parallelize([2,7])并集运算
可以使用union函数进行并集运算：
print (intRDD1.union(intRDD2).union(intRDD3).collect())输出为:
[3, 1, 2, 5, 5, 5, 6, 2, 7]交集运算
可以使用intersection进行交集运算：
print (intRDD1.intersection(intRDD2).collect())两个集合中只有一个相同元素5，所以输出为：
[5]差集运算
可以使用subtract函数进行差集运算：
print (intRDD1.subtract(intRDD2).collect())由于两个RDD的重复部分为5，所以输出为[1,2,3]:
[2, 1, 3]笛卡尔积运算
可以使用cartesian函数进行笛卡尔乘积运算:
print (intRDD1.cartesian(intRDD2).collect())由于两个RDD分别有5个元素和2个元素，所以返回结果有10各元素：
[(3, 5), (3, 6), (1, 5), (1, 6), (2, 5), (2, 6), (5, 5), (5, 6), (5, 5), (5, 6)]4、基本“动作”运算
读取元素
可以使用下列命令读取RDD内的元素，这是Actions运算，所以会马上执行：
#取第一条数据print (intRDD.first())#取前两条数据print (intRDD.take(2))#升序排列，并取前3条数据print (intRDD.takeOrdered(3))#降序排列，并取前3条数据print (intRDD.takeOrdered(3,lambda x:-x))输出为：
3[3, 1]
[1, 2, 3]
[5, 5, 3]统计功能
可以将RDD内的元素进行统计运算：
#统计print (intRDD.stats())#最小值print (intRDD.min())#最大值print (intRDD.max())#标准差print (intRDD.stdev())#计数print (intRDD.count())#求和print (intRDD.sum())#平均print (intRDD.mean())输出为：
(count: 5, mean: 3.2, stdev: 1.6, max: 5, min: 1)151.65163.25、RDD Key-Value基本“转换”运算
Spark RDD支持键值对运算，Key-Value运算时mapreduce运算的基础，本节介绍RDD键值的基本“转换”运算。
初始化
我们用元素类型为tuple元组的数组初始化我们的RDD，这里，每个tuple的第一个值将作为键，而第二个元素将作为值。
kvRDD1 = sc.parallelize([(3,4),(3,6),(5,6),(1,2)])得到key和value值
可以使用keys和values函数分别得到RDD的键数组和值数组：
print (kvRDD1.keys().collect())
print (kvRDD1.values().collect())输出为：
[3, 3, 5, 1]
[4, 6, 6, 2]筛选元素
可以按照键进行元素筛选，也可以通过值进行元素筛选，和之前的一样，使用filter函数，这里要注意的是，虽然RDD中是以键值对形式存在，但是本质上还是一个二元组，二元组的第一个值代表键，第二个值代表值，所以按照如下的代码既可以按照键进行筛选，我们筛选键值小于5的数据：
print (kvRDD1.filter(lambda x:x[0] < 5).collect())输出为：
[(3, 4), (3, 6), (1, 2)]同样，将x[0]替换为x[1]就是按照值进行筛选，我们筛选值小于5的数据：
print (kvRDD1.filter(lambda x:x[1] < 5).collect())输出为：
[(3, 4), (1, 2)]值运算
我们可以使用mapValues方法处理value值，下面的代码将value值进行了平方处理：
print (kvRDD1.mapValues(lambda x:x**2).collect())输出为：
[(3, 16), (3, 36), (5, 36), (1, 4)]按照key排序
可以使用sortByKey按照key进行排序，传入参数的默认值为true，是按照从小到大排序，也可以传入参数false，表示从大到小排序：
print (kvRDD1.sortByKey().collect())
print (kvRDD1.sortByKey(True).collect())
print (kvRDD1.sortByKey(False).collect())输出为：
[(1, 2), (3, 4), (3, 6), (5, 6)]
[(1, 2), (3, 4), (3, 6), (5, 6)]
[(5, 6), (3, 4), (3, 6), (1, 2)]合并相同key值的数据
使用reduceByKey函数可以对具有相同key值的数据进行合并。比如下面的代码，由于RDD中存在（3,4）和（3,6）两条key值均为3的数据，他们将被合为一条数据：
print (kvRDD1.reduceByKey(lambda x,y:x+y).collect())输出为
[(1, 2), (3, 10), (5, 6)]6、多个RDD Key-Value“转换”运算
初始化
首先我们初始化两个k-v的RDD：
kvRDD1 = sc.parallelize([(3,4),(3,6),(5,6),(1,2)])
kvRDD2 = sc.parallelize([(3,8)])内连接运算
join运算可以实现类似数据库的内连接，将两个RDD按照相同的key值join起来，kvRDD1与kvRDD2的key值唯一相同的是3，kvRDD1中有两条key值为3的数据（3,4）和（3,6），而kvRDD2中只有一条key值为3的数据（3,8），所以join的结果是（3，（4,8）） 和（3，（6，8））：
print (kvRDD1.join(kvRDD2).collect())输出为:
[(3, (4, 8)), (3, (6, 8))]左外连接
使用leftOuterJoin可以实现类似数据库的左外连接，如果kvRDD1的key值对应不到kvRDD2，就会显示None
print (kvRDD1.leftOuterJoin(kvRDD2).collect())输出为:
[(1, (2, None)), (3, (4, 8)), (3, (6, 8)), (5, (6, None))]右外连接
使用rightOuterJoin可以实现类似数据库的右外连接，如果kvRDD2的key值对应不到kvRDD1，就会显示None
print (kvRDD1.rightOuterJoin(kvRDD2).collect())输出为：
[(3, (4, 8)), (3, (6, 8))]删除相同key值数据
使用subtractByKey运算会删除相同key值得数据：
print (kvRDD1.subtractByKey(kvRDD2).collect())结果为：
[(1, 2), (5, 6)]7、Key-Value“动作”运算
读取数据
可以使用下面的几种方式读取RDD的数据：
#读取第一条数据print (kvRDD1.first())#读取前两条数据print (kvRDD1.take(2))#读取第一条数据的key值print (kvRDD1.first()[0])#读取第一条数据的value值print (kvRDD1.first()[1])输出为:
(3, 4)
[(3, 4), (3, 6)]34按key值统计：
使用countByKey函数可以统计各个key值对应的数据的条数：
print (kvRDD1.countByKey().collect())输出为：
defaultdict(<type 'int'>, {1: 1, 3: 2, 5: 1})lookup查找运算
使用lookup函数可以根据输入的key值来查找对应的Value值：
print (kvRDD1.lookup(3))输出为：
[4, 6]8、持久化操作
spark RDD的持久化机制，可以将需要重复运算的RDD存储在内存中，以便大幅提升运算效率，有两个主要的函数：
持久化
使用persist函数对RDD进行持久化：
kvRDD1.persist()在持久化的同时我们可以指定持久化存储等级：
等级 | 说明 |
|---|---|
MEMORY_ONLY | 以反序列化的JAVA对象的方式存储在JVM中. 如果内存不够， RDD的一些分区将不会被缓存， 这样当再次需要这些分区的时候，将会重新计算。这是默认的级别。 |
MEMORY_AND_DISK | 以反序列化的JAVA对象的方式存储在JVM中. 如果内存不够， RDD的一些分区将将会缓存在磁盘上，再次需要的时候从磁盘读取。 |
MEMORY_ONLY_SER | 以序列化JAVA对象的方式存储 (每个分区一个字节数组). 相比于反序列化的方式,这样更高效的利用空间， 尤其是使用快速序列化时。但是读取是CPU操作很密集。 |
MEMORY_AND_DISK_SER | 与MEMORY_ONLY_SER相似, 区别是但内存不足时，存储在磁盘上而不是每次重新计算。 |
DISK_ONLY | 只存储RDD在磁盘 |
MEMORY_ONLY_2, MEMORY_AND_DISK_2, etc. | 与上面的级别相同，只不过每个分区的副本只存储在两个集群节点上。 |
OFF_HEAP (experimental) | 将RDD以序列化的方式存储在 Tachyon. 与 MEMORY_ONLY_SER相比, OFF_HEAP减少了垃圾回收。允许执行体更小通过共享一个内存池。因此对于拥有较大堆内存和高并发的环境有较大的吸引力。更重要的是，因为RDD存储在Tachyon上，执行体的崩溃不会造成缓存的丢失。在这种模式下.Tachyon中的内存是可丢弃的，这样 Tachyon 对于从内存中挤出的块不会试图重建它。如果你打算使用Tachyon作为堆缓存，Spark提供了与Tachyon相兼容的版本。 |
首先我们导入相关函数：
from pyspark.storagelevel import StorageLevel在scala中可以直接使用上述的持久化等级关键词，但是在pyspark中封装为了一个类， StorageLevel类，并在初始化时指定一些参数，通过不同的参数组合，可以实现上面的不同存储等级。StorageLevel类的初始化函数如下：
def __init__(self, useDisk, useMemory, useOffHeap, deserialized, replication=1): self.useDisk = useDisk self.useMemory = useMemory self.useOffHeap = useOffHeap self.deserialized = deserialized self.replication = replication那么不同的存储等级对应的参数为:
StorageLevel.DISK_ONLY = StorageLevel(True, False, False, False)
StorageLevel.DISK_ONLY_2 = StorageLevel(True, False, False, False, 2)
StorageLevel.MEMORY_ONLY = StorageLevel(False, True, False, False)
StorageLevel.MEMORY_ONLY_2 = StorageLevel(False, True, False, False, 2)
StorageLevel.MEMORY_AND_DISK = StorageLevel(True, True, False, False)
StorageLevel.MEMORY_AND_DISK_2 = StorageLevel(True, True, False, False, 2)
StorageLevel.OFF_HEAP = StorageLevel(True, True, True, False, 1)"""
.. note:: The following four storage level constants are deprecated in 2.0, since the records \
will always be serialized in Python.
"""StorageLevel.MEMORY_ONLY_SER = StorageLevel.MEMORY_ONLY""".. note:: Deprecated in 2.0, use ``StorageLevel.MEMORY_ONLY`` instead."""StorageLevel.MEMORY_ONLY_SER_2 = StorageLevel.MEMORY_ONLY_2""".. note:: Deprecated in 2.0, use ``StorageLevel.MEMORY_ONLY_2`` instead."""StorageLevel.MEMORY_AND_DISK_SER = StorageLevel.MEMORY_AND_DISK""".. note:: Deprecated in 2.0, use ``StorageLevel.MEMORY_AND_DISK`` instead."""StorageLevel.MEMORY_AND_DISK_SER_2 = StorageLevel.MEMORY_AND_DISK_2""".. note:: Deprecated in 2.0, use ``StorageLevel.MEMORY_AND_DISK_2`` instead."""取消持久化
使用unpersist函数对RDD进行持久化：
kvRDD1.unpersist()9、整理回顾
哇，有关pyspark的RDD的基本操作就是上面这些啦，想要了解更多的盆友们可以参照官网给出的官方文档：http://spark.apache.org/docs/latest/api/python/pyspark.html#pyspark.RDD
今天主要介绍了两种RDD，基本的RDD和Key-Value形式的RDD，介绍了他们的几种“转换”运算和“动作”运算，整理如下：
RDD运算 | 说明 |
|---|---|
基本RDD“转换”运算 | map（对各数据进行转换），filter（过滤符合条件的数据），distinct（去重运算），randomSplit（根据指定的比例随机分为N各RDD），groupBy（根据条件对数据进行分组），union（两个RDD取并集），intersection（两个RDD取交集），subtract（两个RDD取差集）。cartesian（两个RDD进行笛卡尔积运算） |
基本RDD“动作”运算 | first（取第一条数据），take（取前几条数据），takeOrdered（排序后取前N条数据），统计函数 |
Key-Value形式 RDD“转换”运算 | filter（过滤符合条件的数据），mapValues（对value值进行转换），sortByKey（根据key值进行排序），reduceByKey（合并相同key值的数据），join（内连接两个KDD），leftOuterJoin（左外连接两个KDD），rightOuterJoin（右外连接两个RDD），subtractByKey（相当于key值得差集运算） |
Key-Value形式 RDD“动作”运算 | first（取第一条数据），take（取前几条数据），countByKey（根据key值分组统计），lookup（根据key值查找value值） |
RDD持久化 | persist用于对RDD进行持久化，unpersist取消RDD的持久化，注意持久化的存储等级 |
想了解更多？ 那就赶紧来关注我们