百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT技术 > 正文

SparkSQL——DataFrame的创建与使用

wptr33 2025-04-06 22:17 17 浏览

本文主要从以下几个方面介绍SparkSQL中的DataFrame:

第一,SparkSQL的作用

第二,什么是DataFrame

第三,DataFrame与RDD的区别

第四,DataFrame的创建与使用 (Spark1.x与Spark2.x两种不同版本)


第一,SparkSQL的作用

SparkSQL是spark处理结构化数据的一个模块,它的前身是shark,与基础的spark RDD不同,SparkSQL提供了结构化数据及计算结果等信息的接口,在内部,SparkSQL使用这个额外的信息去执行额外的优化,有几种方式可以跟SparkSQL进行交互,包括SQL和DataSet API,使用相同的执行引擎进行计算的时候,无论是使用哪一种计算引擎都可以快速的计算。

在使用RDD进行处理时,需要了解RDD的每个算子的特点,以求得高效的执行相应的操作。在使用SparkSQL时,程序会自动优化算子的执行过程以及使用哪个算子,提高效率。SparkSQL主要用于进行结构化数据的处理,作为分布式的SQL查询引擎。


第二,什么是DataFrame

DataFrame的前身是SchemaRDD,从Spark 1.3.0开始SchemaRDD更名为DataFrame。与SchemaRDD的主要区别是:DataFrame不再直接继承自RDD,而是自己实现了RDD的绝大多数功能。你仍旧可以在DataFrame上调用rdd方法将其转换为一个RDD。DataFrame是一种以RDD为基础的分布式数据集,类似于传统数据库的二维表格,DataFrame带有Schema元信息,即DataFrame所表示的二维表数据集的每一列都带有名称和类型。

1.DataFrame的作用:

是Spark SQL提供的最核心的编程抽象。

以列的形式组织的,分布式的数据集合。

它其实和关系型数据库中的表非常类似,但是底层做了很多的优化。

2.DataFrame可以构建的来源:

(1)结构化的数据文件

(2)Hive中的表

(3)外部的关系型数据库

(4)RDD

第三,DataFrame与RDD的区别

RDD可看作是分布式的对象的集合,Spark并不知道对象的详细模式信息,DataFrame可看作是分布式的Row对象的集合,其提供了由列组成的详细模式信息,使得Spark SQL可以进行某些形式的执行优化。DataFrame和普通的RDD的逻辑框架区别如下所示:


左侧的RDD[Person]虽然以Person为类型参数,但Spark框架本身不了解 Person类的内部结构。

而右侧的DataFrame却提供了详细的结构信息,使得Spark SQL可以清楚地知道该数据集中包含哪些列,每列的名称和类型各是什么。DataFrame多了数据的结构信息,即schema。这样看起来就像一张表了,DataFrame还配套了新的操作数据的方法,DataFrame API(如df.select())和SQL(select id, name from xx_table where ...)。

此外DataFrame还引入了off-heap,意味着JVM堆以外的内存, 这些内存直接受操作系统管理(而不是JVM)。Spark能够以二进制的形式序列化数据(不包括结构)到off-heap中, 当要操作数据时, 就直接操作off-heap内存. 由于Spark理解schema, 所以知道该如何操作。

RDD是分布式的Java对象的集合。DataFrame是分布式的Row对象的集合。DataFrame除了提供了比RDD更丰富的算子以外,更重要的特点是提升执行效率、减少数据读取以及执行计划的优化。

有了DataFrame这个高一层的抽象后,我们处理数据更加简单了,甚至可以用SQL来处理数据了,对开发者来说,易用性有了很大的提升。

不仅如此,通过DataFrame API或SQL处理数据,会自动经过Spark 优化器(Catalyst)的优化,即使你写的程序或SQL不高效,也可以运行的很快。

1、DataFrame与RDD的优缺点

RDD的优缺点:

优点:

(1)编译时类型安全

编译时就能检查出类型错误

(2)面向对象的编程风格

直接通过对象调用方法的形式来操作数据

缺点:

(1)序列化和反序列化的性能开销

无论是集群间的通信, 还是IO操作都需要对对象的结构和数据进行序列化和反序列化。

关于序列化问题可以参考这篇文章:
https://www.toutiao.com/i6846761221337809419/

(2)GC的性能开销

频繁的创建和销毁对象, 势必会增加GC

DataFrame通过引入schema和off-heap(不在堆里面的内存,指的是除了不在堆的内存,使用操作系统上的内存),解决了RDD的缺点, Spark通过schame就能够读懂数据, 因此在通信和IO时就只需要序列化和反序列化数据, 而结构的部分就可以省略了;通过off-heap引入,可以快速的操作数据,避免大量的GC。但是却丢了RDD的优点,DataFrame不是类型安全的, API也不是面向对象风格的。

第四,DataFrame的创建与使用

spark1.x版本

在该版本下,我选择使用的scala2.10.7版本

spark1.x版本中总共有三种编程模式:

1、RDD数据类型调用toDF方法将RDD转化为DataFrame

package xxx

import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.rdd.RDD
import org.apache.spark.sql.{DataFrame, SQLContext}

/**
 * Spark1.x 编程模式
 * 1、创建SparkContext,然后再创建SQLContext
 * 2、先创建RDD,对数据进行整理,然后关联case class,将非结构化数据转化为结构化数据
 * 3、显示的调用toDF方法将RDD转化为DataFrame
 * 4、注册临时表
 * 5、执行SQL(Transformation方式)
 * 6、执行action
 */
object SqlDemo1 {
  def main(args: Array[String]): Unit = {
    //提交的这个程序可以连接到Spark集群中
    val conf = new SparkConf().setAppName("SQLDemo1").setMaster("local[2]")

    //创建SparkSQL的连接(程序执行的入口)
    // sparkContext不能创建特殊的RDD(DataFrame)
    val sc = new SparkContext(conf)

    //将SparkContext包装进而增强
    // 创建特殊的RDD(DataFrame),就是有schema信息的RDD
    val sqlContext = new SQLContext(sc)

    //先有一个普通的RDD,然后在关联上schema,进而转成DataFrame
    val lines = sc.textFile("hdfs://master:9000/test/sparkSQL/person.txt")

    //将数据进行整理
    val boyRDD: RDD[Person] = lines.map(line => {
      val fields = line.split(",")
      val id = fields(0).toInt
      val name = fields(1)
      val age = fields(2).toInt
      val score = fields(3).toInt
      Person(id, name, age, score)
    })

    //该RDD装的是Person类型的数据,有了shcma信息,但是还是一个RDD
    //将RDD转换成DataFrame
    //导入隐式转换
    import sqlContext.implicits._
    val bdf: DataFrame = boyRDD.toDF

    //变成DF后就可以使用两种API进行编程了
    //把DataFrame先注册临时表
    bdf.registerTempTable("t_boy")

    //书写SQL(SQL方法应其实是Transformation)
    val result: DataFrame = sqlContext.sql("SELECT * FROM t_boy ORDER BY score desc, age asc")

    //查看结果(触发Action)
    result.show()

    sc.stop()

  }

  case class Person(id: Int, name: String, age: Int, score: Int)
}

2、在RDD中关联Row,将非结构化的数据转化为结构化的数据,然后定义模式,通过调用SqlContext的createDataFrame方法将RDD转化为DataFrame

package xxx

import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.rdd.RDD
import org.apache.spark.sql.types.{IntegerType, StructField, StructType, StringType}
import org.apache.spark.sql.{DataFrame, Row, SQLContext, types}

/**
 * Spark1.x 编程模式
 * 1、创建SparkContext,然后再创建SQLContext
 * 2、先创建RDD,对数据进行整理,然后关联ROW,将非结构化数据转化为结构化数据
 * 3、定义schema
 * 4、调用sqlContext的createDataFrame方法
 * 5、注册临时表
 * 6、执行SQL(Transformation方式)
 * 7、执行action
 */
object SqlDemo2 {
  def main(args: Array[String]): Unit = {
    //提交的这个程序可以连接到Spark集群中
    val conf = new SparkConf().setAppName("SQLDemo1").setMaster("local[2]")

    //创建SparkSQL的连接(程序执行的入口)
    // sparkContext不能创建特殊的RDD(DataFrame)
    val sc = new SparkContext(conf)

    //将SparkContext包装进而增强
    // 创建特殊的RDD(DataFrame),就是有schema信息的RDD
    val sqlContext = new SQLContext(sc)

    //先有一个普通的RDD,然后在关联上schema,进而转成DataFrame
    val lines = sc.textFile("hdfs://master:9000/test/sparkSQL/person.txt")

    //将数据进行整理
    val rowRDD = lines.map(line => {
      val fields = line.split(",")
      val id = fields(0).toInt
      val name = fields(1)
      val age = fields(2).toInt
      val score = fields(3).toInt
      Row(id, name, age, score)
    })

    // 设置结构类型,表头信息
    val structType = StructType(List(
      StructField("id", IntegerType, true),
      StructField("name", StringType, true),
      StructField("age", IntegerType, true),
      StructField("score", IntegerType, true)
    ))

    // 将RowRDD关联schema
    val bdf = sqlContext.createDataFrame(rowRDD, structType)

    //变成DF后就可以使用两种API进行编程了
    //把DataFrame先注册临时表
    bdf.registerTempTable("t_boy")

    //书写SQL(SQL方法应其实是Transformation)
    val result: DataFrame = sqlContext.sql("SELECT * FROM t_boy ORDER BY score desc, age asc")

    //查看结果(触发Action)
    result.show()

    sc.stop()

  }

}

3、创建DataFrame的方式同1,只是在使用DataFrame的时候不使用SQL语句,而是使用DataFrame API,这样就不需要将DataFrame注册成临时表

package xxx

import org.apache.spark.sql.types.{IntegerType, StringType, StructField, StructType}
import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.sql.{Row, SQLContext}

/**
 * Spark1.x 编程模式
 * 1、创建SparkContext,然后再创建SQLContext
 * 2、先创建RDD,对数据进行整理,然后关联ROW,将非结构化数据转化为结构化数据
 * 3、定义schema
 * 4、调用sqlContext的createDataFrame方法
 * 6、执行DataFrame语句(Transformation方式)
 * 7、执行action
 */
object SqlDemo3 {
  def main(args: Array[String]): Unit = {
    //提交的这个程序可以连接到Spark集群中
    val conf = new SparkConf().setAppName("SQLDemo1").setMaster("local[2]")

    //创建SparkSQL的连接(程序执行的入口)
    // sparkContext不能创建特殊的RDD(DataFrame)
    val sc = new SparkContext(conf)

    //将SparkContext包装进而增强
    // 创建特殊的RDD(DataFrame),就是有schema信息的RDD
    val sqlContext = new SQLContext(sc)

    //先有一个普通的RDD,然后在关联上schema,进而转成DataFrame
    val lines = sc.textFile("hdfs://master:9000/test/sparkSQL/person.txt")

    //将数据进行整理
    val rowRDD = lines.map(line => {
      val fields = line.split(",")
      val id = fields(0).toInt
      val name = fields(1)
      val age = fields(2).toInt
      val score = fields(3).toInt
      Row(id, name, age, score)
    })

    // 设置结构类型,表头信息
    val structType = StructType(List(
      StructField("id", IntegerType, true),
      StructField("name", StringType, true),
      StructField("age", IntegerType, true),
      StructField("score", IntegerType, true)
    ))

    // 将RowRDD关联schema
    val bdf = sqlContext.createDataFrame(rowRDD, structType)

    // 不使用SQL的方式,就不需要注册临时表
    val frame = bdf.select("id", "name", "score")


    import  sqlContext.implicits._
    val frame1 = bdf.orderBy($"score" desc, $"age" asc)

    frame1.show()

    sc.stop()
  }

}

在Spark2.x版本中,引入了SparkSession的概念,它为用户提供了一个统一的切入点来使用Spark的各项功能,用户不但可以使用DataFrame的各种API和可以使用SQL,学习Spark2的难度也会大大降低。SparkConf、SparkContext和SQLContext都已经被封装在SparkSession当中。

在该版本下,我选择使用的scala2.11.12版本

1、创建SparkSession,由于SparkContext被封装在SparkSession中,直接调用
SparkSession.SparkContext.textFile读取文件,在RDD中关联Row,将非结构化的数据转化为结构化的数据,然后定义模式,通过调用SqlContext的createDataFrame方法将RDD转化为DataFrame.

import org.apache.spark.rdd.RDD
import org.apache.spark.sql.types.{DoubleType, IntegerType, StringType, StructField, StructType}
import org.apache.spark.sql.{DataFrame, Dataset, Row, SparkSession}

/**
 * Spark2.x 编程模式
 * 1、创建SparkSession,SparkContext被封装在SparkSession中,直接调用读取文件
 * 2、先创建RDD,对数据进行整理,然后关联ROW,将非结构化数据转化为结构化数据
 * 3、定义schema
 * 4、调用SparkSession的createDataFrame方法
 * 5、注册临时表
 * 6、执行DataFrame语句(Transformation方式)
 * 7、执行action
 */
object SparkTest1 {
  def main(args: Array[String]): Unit = {

    //spark2.x SQL的编程API(SparkSession)
    //是spark2.x SQL执行的入口
    val session = SparkSession.builder()
      .appName("SQLTest1")
      .master("local[*]")
      .getOrCreate()

    //创建RDD
    val lines: RDD[String] = session.sparkContext.textFile("hdfs://master:9000/test/sparkSQL/person.txt")

    //将数据进行整理
    val rowRDD: RDD[Row] = lines.map(line => {
      val fields = line.split(",")
      val id = fields(0).toInt
      val name = fields(1)
      val age = fields(2).toInt
      val score = fields(3).toDouble
      Row(id, name, age, score)
    })

    //结果类型,其实就是表头,用于描述DataFrame
    val schema: StructType = StructType(List(
      StructField("id", IntegerType, true),
      StructField("name", StringType, true),
      StructField("age", IntegerType, true),
      StructField("score", DoubleType, true)
    ))

    //创建DataFrame
    val df: DataFrame = session.createDataFrame(rowRDD, schema)

    // 导入隐式转化
    import session.implicits._
    val df2: Dataset[Row] = df.where($"score" > 98).orderBy($"score" desc, $"age" asc)

    df2.show()

    session.stop()
  }

}

2、创建DataFrame后,同样也可以调用createTempView将DataFrame注册成视图,在视图里使用SQL进行查询。如上面的一样,省略。。。。

相关推荐

Linux高性能服务器设计

C10K和C10M计算机领域的很多技术都是需求推动的,上世纪90年代,由于互联网的飞速发展,网络服务器无法支撑快速增长的用户规模。1999年,DanKegel提出了著名的C10问题:一台服务器上同时...

独立游戏开发者常犯的十大错误

...

学C了一头雾水该咋办?

学C了一头雾水该怎么办?最简单的方法就是你再学一遍呗。俗话说熟能生巧,铁杵也能磨成针。但是一味的为学而学,这个好像没什么卵用。为什么学了还是一头雾水,重点就在这,找出为什么会这个样子?1、概念理解不深...

C++基础语法梳理:inline 内联函数!虚函数可以是内联函数吗?

上节我们分析了C++基础语法的const,static以及this指针,那么这节内容我们来看一下inline内联函数吧!inline内联函数...

C语言实战小游戏:井字棋(三子棋)大战!文内含有源码

井字棋是黑白棋的一种。井字棋是一种民间传统游戏,又叫九宫棋、圈圈叉叉、一条龙、三子旗等。将正方形对角线连起来,相对两边依次摆上三个双方棋子,只要将自己的三个棋子走成一条线,对方就算输了。但是,有很多时...

C++语言到底是不是C语言的超集之一

C与C++两个关系亲密的编程语言,它们本质上是两中语言,只是C++语言设计时要求尽可能的兼容C语言特性,因此C语言中99%以上的功能都可以使用C++完成。本文探讨那些存在于C语言中的特性,但是在C++...

在C++中,如何避免出现Bug?

C++中的主要问题之一是存在大量行为未定义或对程序员来说意外的构造。我们在使用静态分析器检查各种项目时经常会遇到这些问题。但正如我们所知,最佳做法是在编译阶段尽早检测错误。让我们来看看现代C++中的一...

ESL-通过事件控制FreeSWITCH

通过事件提供的最底层控制机制,允许我们有效地利用工具箱,适时选择使用其中的单个工具。FreeSWITCH是一个核心交换与混合矩阵,它周围有几十个模块提供各种功能特性。我们完全控制了所有的即时信息,这些...

物理老师教你学C++语言(中篇)

一、条件语句与实验判断...

C语言入门指南

当然!以下是关于C语言入门编程的基础介绍和入门建议,希望能帮你顺利起步:C语言入门指南...

C++选择结构,让程序自动进行决策

什么是选择结构?正常的程序都是从上至下顺序执行,这就是顺序结构...

C++特性使用建议

1.引用参数使用引用替代指针且所有不变的引用参数必须加上const。在C语言中,如果函数需要修改变量的值,参数必须为指针,如...

C++程序员学习Zig指南(中篇)

1.复合数据类型结构体与方法的对比C++类:...

研一自学C++啃得动吗?

研一自学C++啃得动吗?在开始前我有一些资料,是我根据网友给的问题精心整理了一份「C++的资料从专业入门到高级教程」,点个关注在评论区回复“888”之后私信回复“888”,全部无偿共享给大家!!!个人...

C++关键字介绍

下表列出了C++中的常用关键字,这些关键字不能作为变量名或其他标识符名称。1、autoC++11的auto用于表示变量的自动类型推断。即在声明变量的时候,根据变量初始值的类型自动为此变量选择匹配的...