Spark SQL实战(04)-API编程之DataFrame

简介: 1 SparkSessionSpark Core: SparkContextSpark SQL: 难道就没有SparkContext?

1 SparkSession

Spark Core: SparkContext


Spark SQL: 难道就没有SparkContext?


2.x之后统一的


package com.javaedge.bigdata.chapter04

import org.apache.spark.sql.{DataFrame, SparkSession}

object SparkSessionApp {

 def main(args: Array[String]): Unit = {

   // DF/DS编程的入口点

   val spark: SparkSession = SparkSession.builder()

     .master("local").getOrCreate()

   // 读取文件的API

   val df: DataFrame = spark.read.text("/Users/javaedge/Downloads/sparksql-train/data/input.txt")

   // TODO 业务逻辑处理,通过DF/DS提供的API完成业务

   df.printSchema()

   df.show() // 展示出来  只有一个字段,string类型的value

   spark.stop()

 }

}


1.x的Spark SQL编程入口点


SQLContext

HiveContext

Spark SQL中,SQLContext、HiveContext都是用来创建DataFrame和Dataset主要入口点,二者区别如下:


数据源支持:SQLContext支持的数据源包括JSON、Parquet、JDBC等等,而HiveContext除了支持SQLContext的数据源外,还支持Hive的数据源。因此,如果需要访问Hive中的数据,需要使用HiveContext。

元数据管理:SQLContext不支持元数据管理,因此无法在内存中创建表和视图,只能直接读取数据源中的数据。而HiveContext可以在内存中创建表和视图,并将其存储在Hive Metastore中。

SQL语言支持:SQLContext和HiveContext都支持Spark SQL中的基本语法,例如SELECT、FROM、WHERE等等。但HiveContext还支持Hive中的所有SQL语法,例如INSERT、CREATE TABLE AS等等。

数据格式支持:HiveContext支持更多的数据格式,包括ORC、Avro、SequenceFile等等。而SQLContext只支持JSON、Parquet、JDBC等几种常用的数据格式。

如若访问Hive中数据或在内存中创建表和视图,推荐HiveContext;若只需访问常见数据源,使用SQLContext。


Spark 2.x后,HiveContext已被SparkSession替代,因此推荐SparkSession创建DataFrame、Dataset。


package com.javaedge.bigdata.chapter04

import org.apache.spark.{SparkConf, SparkContext}

import org.apache.spark.sql.{DataFrame, SQLContext}

/**

* 了解即可,已过时

*/

object SQLContextApp {

 def main(args: Array[String]): Unit = {

   val sparkConf: SparkConf = new SparkConf().setAppName("SQLContextApp").setMaster("local")

   // 此处一定要把SparkConf传进来

   val sc: SparkContext = new SparkContext(sparkConf)

   val sqlContext: SQLContext = new SQLContext(sc)

   val df: DataFrame = sqlContext.read.text("/Users/javaedge/Downloads/sparksql-train/data/input.txt")

   df.show()

   sc.stop()

 }

}



output:

+-------------+

|        value|

+-------------+

|     pk,pk,pk|

|jepson,jepson|

|     xingxing|

+-------------+


2 DataFrame

最早在R语言数据分析包中提出,表示一种类似表格的数据结构,其中行和列都可以有命名。


Spark的DataFrame是基于RDD(弹性分布式数据集)的一种高级抽象,类似关系型数据库的表格。Spark DataFrame可看作带有模式(Schema)的RDD,而Schema则是由结构化数据类型(如字符串、整型、浮点型等)和字段名组成。


2.1 命名变迁

Spark 1.0的Spark SQL的数据结构称为SchemaRDD,具有结构化模式(schema)的分布式数据集合。


Spark 1.3版本开始,SchemaRDD重命名为DataFrame,以更好反映其API和功能实质。因此,DataFrame曾被称为SchemaRDD,但现已不再使用这名称。


2.2 Spark SQL的DataFrame优点

可通过SQL语句、API等多种方式进行查询和操作,还支持内置函数、用户自定义函数等功能

支持优化器和执行引擎,可自动对查询计划进行优化,提高查询效率

因此,DataFrame已成Spark SQL核心组件,广泛应用于数据分析、数据挖掘。


3 数据分析选型:PySpark V.S R 语言

数据规模:如果需要处理大型数据集,则使用PySpark更为合适,因为它可以在分布式计算集群上运行,并且能够处理较大规模的数据。而R语言则可能会受限于单机内存和计算能力。

熟练程度:如果你或你的团队已经很熟悉Python,那么使用PySpark也许更好一些,因为你们不需要再去学习新的编程语言。相反,如果已经对R语言很熟悉,那么继续使用R语言也许更为方便。

生态系统:Spark生态系统提供了许多额外的库和工具,例如Spark Streaming和GraphX等,这些库和工具可以与PySpark无缝集成。而R语言的生态系统也有一些类似的库和工具,但相对来说可选择性就更少一些。

总之,选择使用哪种工具进行数据分析应该基于具体情况进行考虑。如果需要处理大规模数据集,并需要与Spark生态系统集成,那么PySpark可能更适合;如果更加熟悉R语言,或者数据量较小,那么使用R语言也可以做到高效的数据分析。


4 深入理解

Dataset是一个分布式数据集,提供RDD强类型和使用强大的lambda函数的能力,并结合了Spark SQL优化的执行引擎。Dataset可以从JVM对象构建而成,并通过函数式转换(如map、flatMap、filter等)进行操作。Scala和Java都支持Dataset API,但Python没有对Dataset API提供支持。由于Python是一种动态语言,许多Dataset API的优点已经自然地可用,例如可以通过名称访问行的字段。R语言也有类似的特点。


DataFrame,具有命名列的Dataset,类似:


关系数据库中的表

Python中的数据框

但内部有更多优化功能。DataFrame可从各种数据源构建,如:


结构化数据文件

Hive表

外部数据库

现有RDD

DataFrame API 在 Scala、Java、Python 和 R 都可用。在Scala和Java中,DataFrame由一组Rows组成的Dataset表示:


Scala API中,DataFrame只是Dataset[Row]的类型别名

Java API中,用户需要使用Dataset表示DataFrame

通常将Scala/Java中的Dataset of Rows称为DataFrame。


5 实战

People.json


{"name":"Michael"}

{"name":"Andy", "age":30}

{"name":"Justin", "age":19}

package com.javaedge.bigdata.chapter04

import org.apache.spark.sql.{DataFrame, SparkSession}

object DataFrameAPIApp {

 def main(args: Array[String]): Unit = {

   val spark = SparkSession.builder()

     .master("local").appName("DataFrameAPIApp")

     .getOrCreate()

   import spark.implicits._

   val people: DataFrame = spark.read.json(

     "/Users/javaedge/Downloads/sparksql-train/data/people.json")

   // 查看DF的内部结构:列名、列的数据类型、是否可以为空

   people.printSchema()

   // 展示出DF内部的数据

   people.show()

 }

}

output:

root

 |-- age: long (nullable = true)

 |-- name: string (nullable = true)

+----+-------+

| age|   name|

+----+-------+

|null|Michael|

|  30|   Andy|

|  19| Justin|

+----+-------+



// DF里面有两列,只要name列 ==> select name from people

// 两个 API 一样的,只是参数不同,使用稍有不同

people.select("name").show()

people.select($"name").show()


output:


+-------+

|   name|

+-------+

|Michael|

|   Andy|

| Justin|

+-------+


// select * from people where age > 21

people.filter($"age" > 21).show()

people.filter("age > 21").show()



output:

+---+----+

|age|name|

+---+----+

| 30|Andy|

+---+----+


// select age, count(1) from people group by age

people.groupBy("age").count().show()


output:

+----+-----+

| age|count|

+----+-----+

|  19|    1|

|null|    1|

|  30|    1|

+----+-----+


createOrReplaceTempView

若现在,我就想完全使用 SQL 查询了,怎么实现 DF 到表的转换呢?


Spark SQL用来将一个 DataFrame 注册成一个临时表(Temporary Table)的方法。之后可使用 Spark SQL 语法及已注册的表名对 DataFrame 进行查询和操作。


允许为 DataFrame 指定一个名称,并将其保存为一个临时表。该表只存在于当前 SparkSession 的上下文,不会在元数据存储中注册表,也不会在磁盘创建任何文件。因此,临时表在SparkSession终止后就会被删。


一旦临时表被注册,就可使用 SQL 或 DSL 对其查询。如:


people.createOrReplaceTempView("people")

spark.sql("select name from people where age > 21").show()


大文件处理


val zips: DataFrame = spark.read.json("/Users/javaedge/Downloads/sparksql-train/data/zips.json")

zips.printSchema()

zips.show(5)

loc信息没用展示全,超过一定长度就使用…来展示,默认只显示前20条:show() ==> show(20) ==> show(numRows, truncate = true)


4.png


不想被截断就这样:

3.png



zips.head(3).foreach(println)

zips.first()

zips.take(5)


Output:


2.png


head(n: Int)

Spark的DataFrame API中的一个方法,可以返回一个包含前n行数据的数组。这个方法通常用于快速检查一个DataFrame的前几行数据,以了解数据集的大致结构和内容。


先对DataFrame使用.limit(n)方法,限制返回行数前n行

然后使用queryExecution方法生成一个Spark SQL查询计划

最后使用collectFromPlan方法收集数据并返回一个包含前n行数据的数组

该 API 可能导致数据集的全部数据被加载到内存,因此在处理大型数据集时应该谨慎使用。若只想查看数据集结构和内容,使用:


.show()

或.take()

获取前几行数据,而非.head()。


// 过滤出大于40000,字段重新命名

zips.filter(zips.col("pop") > 40000)

 .withColumnRenamed("_id", "new_id")

 .show(5, truncate = false)

output:

+------+----------+-----------------------+-----+-----+

|new_id|city      |loc                    |pop  |state|

+------+----------+-----------------------+-----+-----+

|01040 |HOLYOKE   |[-72.626193, 42.202007]|43704|MA   |

|01085 |MONTGOMERY|[-72.754318, 42.129484]|40117|MA   |

|01201 |PITTSFIELD|[-73.247088, 42.453086]|50655|MA   |

|01420 |FITCHBURG |[-71.803133, 42.579563]|41194|MA   |

|01701 |FRAMINGHAM|[-71.425486, 42.300665]|65046|MA   |

+------+----------+-----------------------+-----+-----+



import org.apache.spark.sql.functions._

// 统计加州pop最多的10个城市名称和ID  desc是一个内置函数

zips.select("_id", "city", "pop", "state")

 .filter(zips.col("state") === "CA")

 .orderBy(desc("pop"))

 .show(5, truncate = false)

output:

+-----+------------+-----+-----+

|_id  |city        |pop  |state|

+-----+------------+-----+-----+

|90201|BELL GARDENS|99568|CA   |

|90011|LOS ANGELES |96074|CA   |

|90650|NORWALK     |94188|CA   |

|91331|ARLETA      |88114|CA   |

|90280|SOUTH GATE  |87026|CA   |

+-----+------------+-----+-----+



可惜啊,我不会写代码,可以使用 MySQL 语法吗?


zips.createOrReplaceTempView("zips")

spark.sql("select _id,city,pop,state" +

 "from zips where state='CA'" +

 "order by pop desc" +

 "limit 10").show()


import spark.implicits._ 作用

在Scala中使用Apache Spark进行数据分析时经常用到的,它的作用是将隐式转换函数导入当前作用域中。这些隐式转换函数包含了许多DataFrame和Dataset的转换方法,例如将RDD转换为DataFrame或将元组转换为Dataset等。


具体来说,这行代码使用了SparkSession对象中的implicits属性,该属性返回了一个类型为org.apache.spark.sql.SQLImplicits的实例。通过调用该实例的方法,可以将各种Scala数据类型(如case class、元组等)与Spark SQL中的数据类型(如Row、DataFrame、Dataset等)之间进行转换,从而方便地进行数据操作和查询。


在使用许多Spark SQL API的时候,往往需要使用这行代码将隐式转换函数导入当前上下文,以获得更加简洁和易于理解的代码编写方式。


如果不导入会咋样

如果不导入spark.implicits._会导致编译错误或者运行时异常。因为在进行DataFrame和Dataset的操作时,需要使用到一些隐式转换函数。如果没有导入spark.implicits._,则这些隐式转换函数无法被自动引入当前上下文,就需要手动地导入这些函数,这样会使编码变得比较麻烦。


例如,在进行RDD和DataFrame之间的转换时,如果不导入spark.implicits._,则需要手动导入org.apache.spark.sql.Row以及org.apache.spark.sql.functions._等包,并通过调用toDF()方法将RDD转换为DataFrame。而有了导入spark.implicits._后,只需要直接调用RDD对象的toDF()方法即可完成转换。


因此,为了简化编码,通常会在Scala中使用Spark SQL时导入spark.implicits._,从而获得更加简洁易读的代码。


案例

people.select($"name").show()


如果不导入 spark.implicits._,则可以手动创建一个 Column 对象来进行筛选操作。例如,可以使用 col 函数来创建一个 Column 对象,然后在 select 方法中使用该列:


import org.apache.spark.sql.functions.col

val selected = people.select(col("name"))

selected.show()


这样就可以实现与 people.select($"name").show() 相同的效果,但需要手动创建 Column 对象。显然,在编写复杂的数据操作时,手动创建 Column 对象可能会变得非常繁琐和困难,因此通常情况下我们会选择使用隐式转换函数,从而更加方便地使用DataFrame的API。

目录
相关文章
|
3天前
|
SQL 分布式计算 Java
Spark SQL向量化执行引擎框架Gluten-Velox在AArch64使能和优化
本文摘自 Arm China的工程师顾煜祺关于“在 Arm 平台上使用 Native 算子库加速 Spark”的分享,主要内容包括以下四个部分: 1.技术背景 2.算子库构成 3.算子操作优化 4.未来工作
|
2月前
|
SQL JSON 分布式计算
【赵渝强老师】Spark SQL的数据模型:DataFrame
本文介绍了在Spark SQL中创建DataFrame的三种方法。首先,通过定义case class来创建表结构,然后将CSV文件读入RDD并关联Schema生成DataFrame。其次,使用StructType定义表结构,同样将CSV文件读入RDD并转换为Row对象后创建DataFrame。最后,直接加载带有格式的数据文件(如JSON),通过读取文件内容直接创建DataFrame。每种方法都包含详细的代码示例和解释。
|
3月前
|
SQL 分布式计算 大数据
大数据-97 Spark 集群 SparkSQL 原理详细解析 Broadcast Shuffle SQL解析过程(一)
大数据-97 Spark 集群 SparkSQL 原理详细解析 Broadcast Shuffle SQL解析过程(一)
107 0
|
JSON 资源调度 分布式计算
基于Yarn API的Spark程序监控
一.简述 通过对Yarn ResourceManager中运行程序的状态(RUNNING、KILLED、FAILED、FINISHED)以及ApplicationMaster中Application的Job执行时长超过批次时间的监控,来达到对Spark on Yarn程序的失败重启、超时重启等功能 二.
4939 0
|
8天前
|
JSON 前端开发 搜索推荐
关于商品详情 API 接口 JSON 格式返回数据解析的示例
本文介绍商品详情API接口返回的JSON数据解析。最外层为`product`对象,包含商品基本信息(如id、name、price)、分类信息(category)、图片(images)、属性(attributes)、用户评价(reviews)、库存(stock)和卖家信息(seller)。每个字段详细描述了商品的不同方面,帮助开发者准确提取和展示数据。具体结构和字段含义需结合实际业务需求和API文档理解。
|
2天前
|
JSON 搜索推荐 API
京东店铺所有商品接口系列(京东 API)
本文介绍如何使用Python调用京东API获取店铺商品信息。前期需搭建Python环境,安装`requests`库并熟悉`json`库的使用。接口采用POST请求,参数包括`app_key`、`method`、`timestamp`、`v`、`sign`和业务参数`360buy_param_json`。通过示例代码展示如何生成签名并发送请求。应用场景涵盖店铺管理、竞品分析、数据统计及商品推荐系统,帮助商家优化运营和提升竞争力。
34 23
|
13天前
|
JSON API 数据格式
京东商品SKU价格接口(Jd.item_get)丨京东API接口指南
京东商品SKU价格接口(Jd.item_get)是京东开放平台提供的API,用于获取商品详细信息及价格。开发者需先注册账号、申请权限并获取密钥,随后通过HTTP请求调用API,传入商品ID等参数,返回JSON格式的商品信息,包括价格、原价等。接口支持GET/POST方式,适用于Python等语言的开发环境。
66 11
|
5天前
|
Web App开发 JSON 测试技术
API测试工具集合:让接口测试更简单高效
在当今软件开发领域,接口测试工具如Postman、Apifox、Swagger等成为确保API正确性、性能和可靠性的关键。Postman全球闻名但高级功能需付费,Apifox则集成了API文档、调试、Mock与自动化测试,简化工作流并提高团队协作效率,特别适合国内用户。Swagger自动生成文档,YApi开源但功能逐渐落后,Insomnia界面简洁却缺乏团队协作支持,Paw仅限Mac系统。综合来看,Apifox是国内用户的理想选择,提供中文界面和免费高效的功能。
|
1月前
|
人工智能 自然语言处理 API
Multimodal Live API:谷歌推出新的 AI 接口,支持多模态交互和低延迟实时互动
谷歌推出的Multimodal Live API是一个支持多模态交互、低延迟实时互动的AI接口,能够处理文本、音频和视频输入,提供自然流畅的对话体验,适用于多种应用场景。
90 3
Multimodal Live API:谷歌推出新的 AI 接口,支持多模态交互和低延迟实时互动
|
24天前
|
JSON 安全 API
淘宝商品详情API接口(item get pro接口概述)
淘宝商品详情API接口旨在帮助开发者获取淘宝商品的详细信息,包括商品标题、描述、价格、库存、销量、评价等。这些信息对于电商企业而言具有极高的价值,可用于商品信息展示、市场分析、价格比较等多种应用场景。