You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

445 lines
24 KiB
Markdown

This file contains invisible Unicode characters!

This file contains invisible Unicode characters that may be processed differently from what appears below. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to reveal hidden characters.

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

# 16 | 数据转换如何在DataFrame之上做数据处理
你好,我是吴磊。
在上一讲我们学习了创建DataFrame的各种途径与方法那么有了DataFrame之后我们该如何在DataFrame之上做数据探索、数据分析以及各式各样的数据转换呢在数据处理完毕之后我们又该如何做数据展示与数据持久化呢今天这一讲我们就来解答这些疑问。
为了给开发者提供足够的灵活性对于DataFrame之上的数据处理Spark SQL支持两类开发入口一个是大家所熟知的结构化查询语言SQL另一类是DataFrame开发算子。就开发效率与执行效率来说二者并无优劣之分选择哪种开发入口完全取决于开发者的个人偏好与开发习惯。
与RDD类似DataFrame支持种类繁多的开发算子但相比SQL语言DataFrame算子的学习成本相对要高一些。因此本着先易后难的思路咱们先来说说DataFrame中SQL语句的用法然后再去理解DataFrame开发算子。
## SQL语句
对于任意的DataFrame我们都可以使用createTempView或是createGlobalTempView在Spark SQL中创建临时数据表。
两者的区别在于createTempView创建的临时表其生命周期仅限于SparkSession内部而createGlobalTempView创建的临时表可以在同一个应用程序中跨SparkSession提供访问。有了临时表之后我们就可以使用SQL语句灵活地倒腾表数据。
通过后面这段代码我为你演示了如何使用createTempView创建临时表。我们首先用toDF创建了一个包含名字和年龄的DataFrame然后调用createTempView方法创建了临时表。
```scala
import org.apache.spark.sql.DataFrame
import spark.implicits._
 
val seq = Seq(("Alice", 18), ("Bob", 14))
val df = seq.toDF("name", "age")
 
df.createTempView("t1")
val query: String = "select * from t1"
// spark为SparkSession实例对象
val result: DataFrame = spark.sql(query)
 
result.show
 
/** 结果打印
+-----+---+
| n ame| age|
+-----+---+
| Alice| 18|
| Bob| 14|
+-----+---+
*/
```
以上表为例我们先是使用spark.implicits.\_隐式方法通过toDF来创建DataFrame然后在其上调用createTempView来创建临时表“t1”。接下来给定SQL查询语句“query”我们可以通过调用SparkSession提供的sql API来提请执行查询语句得到的查询结果被封装为新的DataFrame。
值得一提的是与RDD的开发模式一样**DataFrame之间的转换也属于延迟计算当且仅当出现Action类算子时如上表中的show所有之前的转换过程才会交付执行**。
Spark SQL采用[ANTLR](https://www.antlr.org/)语法解析器来解析并处理SQL语句。我们知道ANTLR是一款强大的、跨语言的语法解析器因为它全面支持SQL语法所以广泛应用于Oracle、Presto、Hive、ElasticSearch等分布式数据仓库和计算引擎。因此像Hive或是Presto中的SQL查询语句都可以平滑地迁移到Spark SQL。
不仅如此Spark SQL还提供大量Built-in Functions内置函数用于辅助数据处理如array\_distinct、collect\_list等等。你可以浏览官网的[Built-in Functions页面](https://spark.apache.org/docs/3.0.1/api/sql/index.html)查找完整的函数列表。结合SQL语句以及这些灵活的内置函数你就能游刃有余地应对数据探索、数据分析这些典型的数据应用场景。
SQL语句相对比较简单学习路径短、成本低你只要搞清楚如何把DataFrame转化为数据表剩下的事就水到渠成了。接下来我们把主要精力放在DataFrame支持的各类算子上这些算子提供的功能往往能大幅提升开发效率让我们事半功倍。
## DataFrame算子
不得不说DataFrame支持的算子丰富而又全面这主要源于DataFrame特有的“双面”属性。一方面DataFrame来自RDD与RDD具有同源性因此RDD支持的大部分算子DataFrame都支持。另一方面DataFrame携带Schema是结构化数据因此它必定要提供一套与结构化查询同源的计算算子。
正是由于这样“双面”的特性我们从下图可以看到DataFrame所支持的算子用“琳琅满目”来形容都不为过。
![图片](https://static001.geekbang.org/resource/image/50/97/50bd70a2dcf01b631eff86c286d9eb97.jpg?wh=1920x442 "DataFrame算子大全")
人类的大脑偏好结构化的知识为了方便你记忆与理解我把DataFrame上述两个方面的算子进一步划分为6大类**它们分别是RDD同源类算子、探索类算子、清洗类算子、转换类算子、分析类算子和持久化算子**。
你可能会困扰“天呐这么多算子要学这不是逼我从入门到放弃吗”别着急上面这张图你可以把它当作是“DataFrame算子脑图”或是一本字典。在日常的开发中思路枯竭的时候你不妨把它翻出来看看哪些算子能够帮你实现业务逻辑。
今天这一讲,我们也会根据这张“脑图”,重点讲解其中最常用、最关键的部分。
### 同源类算子
我们从DataFrame中的RDD同源类算子说起这些算子在RDD算子那三讲做过详细的介绍如果你对有哪个算子的作用或含义记不清了不妨回看之前的三讲。我按照之前的分类把这些算子整理成了一张表格。
![图片](https://static001.geekbang.org/resource/image/5c/53/5c377yy9b592e2ae1a31cb68da2d5553.jpg?wh=1592x917 "RDD同源类算子")
### 探索类算子
接下来就是DataFrame的探索类算子。所谓探索指的是数据探索这类算子的作用在于帮助开发者初步了解并认识数据比如数据的模式Schema、数据的分布、数据的“模样”等等为后续的应用开发奠定基础。
对于常用的探索类算子,我把它们整理到了下面的表格中,你不妨先看一看,建立“第一印象”。
![图片](https://static001.geekbang.org/resource/image/7c/15/7c95d03be0a733b1498ba4b99f2e1d15.jpg?wh=1588x789 "探索类算子")
我们来依次“避轻就重”地说一说这些算子。首先columns/schema/printSchema这3个算子类似都可以帮我们获取DataFrame的数据列和Schema。尤其是printSchema它以纯文本的方式将Data Schema打印到屏幕上如下所示。
```scala
import org.apache.spark.sql.DataFrame
import spark.implicits._
 
val employees = Seq((1, "John", 26, "Male"), (2, "Lily", 28, "Female"), (3, "Raymond", 30, "Male"))
val employeesDF: DataFrame = employees.toDF("id", "name", "age", "gender")
 
employeesDF.printSchema
 
/** 结果打印
root
|-- id: integer (nullable = false)
|-- name: string (nullable = true)
|-- age: integer (nullable = false)
|-- gender: string (nullable = true)
*/
```
了解数据模式之后我们往往想知道数据具体长什么样子对于这个诉求show算子可以帮忙达成。在默认情况下show会随机打印出DataFrame的20条数据记录。
```scala
employeesDF.show
 
/** 结果打印
+---+-------+---+------+
| id| name|age|gender|
+---+-------+---+------+
| 1| John| 26| Male|
| 2| Lily| 28|Female|
| 3|Raymond| 30| Male|
+---+-------+---+------+
*/
```
看清了数据的“本来面目”之后你还可以进一步利用describe去查看数值列的统计分布。比如通过调用employeesDF.describe(“age”)你可以查看age列的极值、平均值、方差等统计数值。
初步掌握了数据的基本情况之后如果你对当前DataFrame的执行计划感兴趣可以通过调用explain算子来获得Spark SQL给出的执行计划。explain对于执行效率的调优来说有着至关重要的作用后续课程中我们还会结合具体的实例来深入讲解explain的用法和释义在这里你仅需知道explain是用来查看执行计划的就好。
### 清洗类算子
完成数据探索以后我们正式进入数据应用的开发阶段。在数据处理前期我们往往需要对数据进行适当地“清洗”“洗掉”那些不符合业务逻辑的“脏数据”。DataFrame提供了如下算子来帮我们完成这些脏活儿、累活儿。
![图片](https://static001.geekbang.org/resource/image/c1/1b/c1c55259a2f14870606ab4c182d8921b.jpg?wh=1588x789 "清洗类算子")
首先drop算子允许开发者直接把指定列从DataFrame中予以清除。举个例子对于上述的employeesDF假设我们想把性别列清除那么直接调用 employeesDF.drop(“gender”) 即可。如果要同时清除多列只需要在drop算子中用逗号把多个列名隔开即可。
第二个是distinct它用来为DataFrame中的数据做去重。还是以employeesDF为例当有多条数据记录的所有字段值都相同时使用distinct可以仅保留其中的一条数据记录。
接下来是dropDuplicates它的作用也是去重。不过与distinct不同的是dropDuplicates可以指定数据列因此在灵活性上更胜一筹。还是拿employeesDF来举例这个DataFrame原本有3条数据记录如果我们按照性别列去重最后只会留下两条记录。其中一条记录的gender列是“Male”另一条的gender列为“Female”如下所示。
```scala
employeesDF.show
 
/** 结果打印
+---+-------+---+------+
| id| name|age|gender|
+---+-------+---+------+
| 1| John| 26| Male|
| 2| Lily| 28|Female|
| 3|Raymond| 30| Male|
+---+-------+---+------+
*/
 
employeesDF.dropDuplicates("gender").show
 
/** 结果打印
+---+----+---+------+
| id|name|age|gender|
+---+----+---+------+
| 2|Lily| 28|Female|
| 1|John| 26| Male|
+---+----+---+------+
*/
```
表格中的最后一个算子是na它的作用是选取DataFrame中的null数据na往往要结合drop或是fill来使用。例如employeesDF.na.drop用于删除DataFrame中带null值的数据记录而employeesDF.na.fill(0) 则将DataFrame中所有的null值都自动填充为整数零。这两种用例在数据清洗的场景中都非常常见因此你需要牢牢掌握na.drop与na.fill的用法。
数据清洗过后,我们就得到了一份“整洁而又干净”的数据,接下来,可以放心大胆地去做各式各样的数据转换,从而实现业务逻辑需求。
### 转换类算子
转换类算子的主要用于数据的生成、提取与转换。转换类的算子的数量并不多,但使用方式非常灵活,开发者可以变着花样地变换数据。
![图片](https://static001.geekbang.org/resource/image/4a/77/4a3f705c5bde5a597yy5bf8c78b15e77.jpg?wh=1640x1047 "转换类算子")
首先select算子让我们可以按照列名对DataFrame做投影比如说如果我们只关心年龄与性别这两个字段的话就可以使用下面的语句来实现。
```scala
employeesDF.select("name", "gender").show
 
/** 结果打印
+-------+------+
| name|gender|
+-------+------+
| John| Male|
| Lily|Female|
|Raymond| Male|
+-------+------+
*/
```
不过虽然用起来比较简单但select算子在功能方面不够灵活。在灵活性这方面selectExpr做得更好。比如说基于id和姓名我们想把它们拼接起来生成一列新的数据。像这种需求正是selectExpr算子的用武之地。
```scala
employeesDF.selectExpr("id", "name", "concat(id, '_', name) as id_name").show
 
/** 结果打印
+---+-------+---------+
| id| name| id_name|
+---+-------+---------+
| 1| John| 1_John|
| 2| Lily| 2_Lily|
| 3|Raymond|3_Raymond|
+---+-------+---------+
*/
```
这里我们使用concat这个函数把id列和name列拼接在一起生成新的id\_name数据列。
接下来的where和withColumnRenamed这两个算子比较简单where使用SQL语句对DataFrame做数据过滤而withColumnRenamed的作用是字段重命名。
比如想要过滤出所有性别为男的员工我们就可以用employeesDF.where(“gender = Male”)来实现。如果打算把employeesDF当中的“gender”重命名为“sex”就可以用withColumnRenamed来帮忙employeesDF.withColumnRenamed(“gender”, “sex”)。
紧接着的是withColumn虽然名字看上去和withColumnRenamed很像但二者在功能上有着天壤之别。
withColumnRenamed是重命名现有的数据列而withColumn则用于生成新的数据列这一点上withColumn倒是和selectExpr有着异曲同工之妙。withColumn也可以充分利用Spark SQL提供的Built-in Functions来灵活地生成数据。
比如,基于年龄列,我们想生成一列脱敏数据,隐去真实年龄,你就可以这样操作。
```scala
employeesDF.withColumn("crypto", hash($"age")).show
 
/** 结果打印
+---+-------+---+------+-----------+
| id| name|age|gender| crypto|
+---+-------+---+------+-----------+
| 1| John| 26| Male|-1223696181|
| 2| Lily| 28|Female|-1721654386|
| 3|Raymond| 30| Male| 1796998381|
+---+-------+---+------+-----------+
*/
```
可以看到我们使用内置函数hash生成一列名为“crypto”的新数据数据值是对应年龄的哈希值。有了新的数据列之后我们就可以调用刚刚讲的drop把原始的age字段丢弃掉。
表格中的最后一个算子是explode这个算子很有意思它的作用是展开数组类型的数据列数组当中的每一个元素都会生成一行新的数据记录。为了更好地演示explode的用法与效果我们把employeesDF数据集做个简单的调整给它加上一个interests兴趣字段。
```scala
val seq = Seq( (1, "John", 26, "Male", Seq("Sports", "News")),
(2, "Lily", 28, "Female", Seq("Shopping", "Reading")),
(3, "Raymond", 30, "Male", Seq("Sports", "Reading"))
)
 
val employeesDF: DataFrame = seq.toDF("id", "name", "age", "gender", "interests")
employeesDF.show
 
/** 结果打印
+---+-------+---+------+-------------------+
| id| name|age|gender| interests|
+---+-------+---+------+-------------------+
| 1| John| 26| Male| [Sports, News]|
| 2| Lily| 28|Female|[Shopping, Reading]|
| 3|Raymond| 30| Male| [Sports, Reading]|
+---+-------+---+------+-------------------+
*/
 
employeesDF.withColumn("interest", explode($"interests")).show
 
/** 结果打印
+---+-------+---+------+-------------------+--------+
| id| name|age|gender| interests|interest|
+---+-------+---+------+-------------------+--------+
| 1| John| 26| Male| [Sports, News]| Sports|
| 1| John| 26| Male| [Sports, News]| News|
| 2| Lily| 28|Female|[Shopping, Reading]|Shopping|
| 2| Lily| 28|Female|[Shopping, Reading]| Reading|
| 3|Raymond| 30| Male| [Sports, Reading]| Sports|
| 3|Raymond| 30| Male| [Sports, Reading]| Reading|
+---+-------+---+------+-------------------+--------+
*/
```
可以看到,我们多加了一个兴趣列,列数据的类型是数组,每个员工都有零到多个兴趣。
如果我们想把数组元素展开让每个兴趣都可以独占一条数据记录。这个时候就可以使用explode再结合withColumn生成一列新的interest数据。这列数据的类型是单个元素的String而不再是数组。有了新的interest数据列之后我们可以再次利用drop算子把原本的interests列抛弃掉。
数据转换完毕之后我们就可以通过数据的关联、分组、聚合、排序去做数据分析从不同的视角出发去洞察数据。这个时候我们还要依赖Spark SQL提供的多个分析类算子。
### 分析类算子
毫不夸张地说,前面的探索、清洗、转换,都是在为数据分析做准备。**在大多数的数据应用中,数据分析往往是最为关键的那环,甚至是应用本身的核心目的**。因此,熟练掌握分析类算子,有利于我们提升开发效率。
Spark SQL的分析类算子看上去并不多但灵活组合使用就会有“千变万化”的效果让我们一起看看。
![图片](https://static001.geekbang.org/resource/image/6f/1c/6f1b2yy4ceb3e7c5bbyyec25e04e791c.jpg?wh=1564x797 "分析类算子")
为了演示上述算子的用法我们先来准备两张数据表employees和salaries也即员工信息表和薪水表。我们的想法是通过对两张表做数据关联来分析员工薪水的分布情况。
```scala
import spark.implicits._
import org.apache.spark.sql.DataFrame
 
// 创建员工信息表
val seq = Seq((1, "Mike", 28, "Male"), (2, "Lily", 30, "Female"), (3, "Raymond", 26, "Male"))
val employees: DataFrame = seq.toDF("id", "name", "age", "gender")
 
// 创建薪水表
val seq2 = Seq((1, 26000), (2, 30000), (4, 25000), (3, 20000))
val salaries:DataFrame = seq2.toDF("id", "salary")
 
employees.show
 
/** 结果打印
+---+-------+---+------+
| id| name|age|gender|
+---+-------+---+------+
| 1| Mike| 28| Male|
| 2| Lily| 30|Female|
| 3|Raymond| 26| Male|
+---+-------+---+------+
*/
 
salaries.show
 
/** 结果打印
+---+------+
| id|salary|
+---+------+
| 1| 26000|
| 2| 30000|
| 4| 25000|
| 3| 20000|
+---+------+
*/
```
那么首先我们先用join算子把两张表关联起来关联键Join Keys我们使用两张表共有的id列而关联形式Join Type自然是内关联Inner Join
```scala
val jointDF: DataFrame = salaries.join(employees, Seq("id"), "inner")
 
jointDF.show
 
/** 结果打印
+---+------+-------+---+------+
| id|salary| name|age|gender|
+---+------+-------+---+------+
| 1| 26000| Mike| 28| Male|
| 2| 30000| Lily| 30|Female|
| 3| 20000|Raymond| 26| Male|
+---+------+-------+---+------+
*/
```
可以看到我们在salaries之上调用join算子join算子的参数有3类。第一类是待关联的数据表在我们的例子中就是员工表employees。第二类是关联键也就是两张表之间依据哪些字段做关联我们这里是id列。第三类是关联形式我们知道关联形式有inner、left、right、anti、semi等等这些关联形式我们下一讲再展开这里你只需要知道Spark SQL支持这些种类丰富的关联形式即可。
数据完成关联之后,我们实际得到的仅仅是最细粒度的事实数据,也就是每个员工每个月领多少薪水。这样的事实数据本身并没有多少价值,我们往往需要从不同的维度出发,对数据做分组、聚合,才能获得更深入、更有价值的数据洞察。
比方说,我们想以性别为维度,统计不同性别下的总薪水和平均薪水,借此分析薪水与性别之间可能存在的关联关系。
```scala
val aggResult = fullInfo.groupBy("gender").agg(sum("salary").as("sum_salary"), avg("salary").as("avg_salary"))
 
aggResult.show
 
/** 数据打印
+------+----------+----------+
|gender|sum_salary|avg_salary|
+------+----------+----------+
|Female| 30000| 30000.0|
| Male| 46000| 23000.0|
+------+----------+----------+
*/
```
这里我们先是使用groupBy算子按照“gender”列做分组然后使用agg算子做聚合运算。在agg算子中我们分别使用sum和avg聚合函数来计算薪水的总数和平均值。Spark SQL对于聚合函数的支持我们同样可以通过[Built-in Functions页面](https://spark.apache.org/docs/3.0.1/api/sql/)来进行检索。结合Built-in Functions提供的聚合函数我们就可以灵活地对数据做统计分析。
得到统计结果之后为了方便查看我们还可以使用sort或是orderBy算子对结果集进行排序二者在用法与效果上是完全一致的如下表所示。
```scala
aggResult.sort(desc("sum_salary"), asc("gender")).show
 
/** 结果打印
+------+----------+----------+
|gender|sum_salary|avg_salary|
+------+----------+----------+
| Male| 46000| 23000.0|
|Female| 30000| 30000.0|
+------+----------+----------+
*/
 
aggResult.orderBy(desc("sum_salary"), asc("gender")).show
 
/** 结果打印
+------+----------+----------+
|gender|sum_salary|avg_salary|
+------+----------+----------+
| Male| 46000| 23000.0|
|Female| 30000| 30000.0|
+------+----------+----------+
*/
```
可以看到sort / orderBy支持按照多列进行排序且可以通过desc和asc来指定排序方向。其中desc表示降序排序相应地asc表示升序排序。
好啦到此为止我们沿着数据的生命周期分别梳理了生命周期不同阶段的Spark SQL算子它们分别是探索类算子、清洗类算子、转换类算子和分析类算子。
![图片](https://static001.geekbang.org/resource/image/50/97/50bd70a2dcf01b631eff86c286d9eb97.jpg?wh=1920x442 "数据生命周期与DataFrame算子")
所谓行百里者半九十纵观整个生命周期我们还剩下数据持久化这一个环节。对于最后的这个持久化环节Spark SQL提供了write API与上一讲介绍的read API相对应write API允许开发者把数据灵活地物化为不同的文件格式。
### 持久化类算子
没有对比就没有鉴别在学习write API之前我们不妨先来回顾一下上一讲介绍的read API。
![图片](https://static001.geekbang.org/resource/image/52/43/525441865dede68fa5a9138cb930de43.jpg?wh=1920x654 "read API一般用法")
如上图所示read API有3个关键点一是由format指定的文件格式二是由零到多个option组成的加载选项最后一个是由load标记的源文件路径。
与之相对write API也有3个关键环节分别是同样由format定义的文件格式零到多个由option构成的“写入选项”以及由save指定的存储路径如下图所示。
![图片](https://static001.geekbang.org/resource/image/73/cd/7348bdf46f3d051b891620a0c3f22dcd.jpg?wh=1920x680 "write API一般用法")
这里的format和save与read API中的format和load是一一对应的分别用于指定文件格式与存储路径。实际上option选项也是类似的除了mode以外write API中的选项键与read API中的选项键也是相一致的如seq用于指定CSV文件分隔符、dbtable用于指定数据表名、等等你可以通过回顾[上一讲](https://time.geekbang.org/column/article/426101)来获取更多的option选项。
在read API中mode选项键用于指定读取模式如permissive, dropMalformed, failFast。但在write API中mode用于指定“写入模式”分别有Append、Overwrite、ErrorIfExists、Ignore这4种模式它们的含义与描述如下表所示。
![图片](https://static001.geekbang.org/resource/image/f9/48/f9367e4bb06702f396325183014ef448.jpg?wh=1594x789 "不同写入模式的含义")
有了write API我们就可以灵活地把DataFrame持久化到不同的存储系统中为数据的生命周期画上一个圆满的句号。
## 重点回顾
今天这一讲我们主要围绕数据的生命周期学习了Spark SQL在不同数据阶段支持的处理算子如下图所示。
![图片](https://static001.geekbang.org/resource/image/50/97/50bd70a2dcf01b631eff86c286d9eb97.jpg?wh=1920x442 "Spark SQL算子一览")
图中涉及的算子很多尽管大部分我们都举例讲过了但要在短时间之内一下子掌握这么多内容确实强人所难。不过你不用担心今天这一讲最主要的目的还是想让你对Spark SQL支持的算子有一个整体的把握。
至于每个算子具体是用来做什么的,在日后的开发工作中,你可以反复地翻看这一讲,结合实践慢慢地加深印象,这样学习更高效。我也强烈建议你空闲时把官网的[Built-in Functions列表](https://spark.apache.org/docs/3.0.1/api/sql/index.html)过一遍,对这些内置函数的功能做到心中有数,实现业务逻辑时才会手到擒来。
除了DataFrame本身支持的算子之外在功能上SQL完全可以实现同样的数据分析。给定DataFrame你只需通过createTempView或是createGlobalTempView来创建临时表然后就可以通过写SQL语句去进行数据的探索、倾斜、转换与分析。
最后需要指出的是DataFrame算子与SQL查询语句之间并没有优劣之分他们可以实现同样的数据应用而且在执行性能方面也是一致的。因此你可以结合你的开发习惯与偏好自由地在两者之间进行取舍。
## 每课一练
在转换类算子中我们举例介绍了explode这个算子它的作用是按照以数组为元素的数据列把一条数据展开爆炸成多条数据。结合这个算子的作用你能否分析一下explode操作是否会引入Shuffle计算呢
欢迎你在留言区跟我交流互动,也推荐你把这一讲分享给有需要的朋友。