大数据系列篇-spark-sql使用SQL加DSL方式与RDD-DATAFRAME-DATASET转换

该博客演示了如何在Spark中使用SQL和DSL查询DataFrame,以及DataFrame与RDD和DataSet之间的转换。通过案例展示了从JSON文件加载数据到DataFrame,执行SQL查询,应用DSL选择特定列,以及DataFrame与DataSet和RDD之间的相互转化操作。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

大数据系列篇-spark-sql使用SQL加DSL方式与RDD-DATAFRAME-DATASET转换

package com.test

import org.apache.spark.SparkConf
import org.apache.spark.sql.SparkSession

//测试sql与dsl与转换RDD <-> DATAFRAME <-> DATASET
object SparkSql {

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

    val sparkConf = new SparkConf().setAppName("练习SparkSql").setMaster("local[*]")
    val spark = SparkSession.builder().config(sparkConf).getOrCreate()
    import spark.implicits._

    //DataFrame
    val df = spark.read.json("data/user.json")
    df.show()

    df.createOrReplaceTempView("user")
    //SQL
    spark.sql("SELECT * FROM user").show()
    //DSL
    df.select("userName").show()
    df.select($"age" + 1).show()
    //或df.select('age + 1).show()

    //DataSet
    val ds = Seq(1, 5, 6).toDS()
    ds.show()

    //RDD<--->DF
    val rdd1 = spark.sparkContext.makeRDD(List((1, "user1", 1), (2, "user1", 2)))
    val df1 = rdd1.toDF("id", "name", "age")
    df1.rdd

    //DF<--->DS
    val ds1 = df1.as[User]
    val df2 = ds1.toDF()

    //RDD<--->DS
    val ds2 = rdd1.map {
      case (id, name, age) => {
        User(id, name, age)
      }
    }.toDS()

    ds2.rdd

    spark.close()

  }

  case class User(id: Long, name: String, age: Int)
}
<think>首先,用户的问题是:“我想理解SparkDataFrameDataset的区别 请问Apache Spark DataFrame vs Dataset difference”。我需要基于提供的引用信息来回答这个问题。 回顾系统级指令: - 所有行内数学表达式必须使用$...$格式。 - 独立公式使用$$...$$格式。 - LaTeX语法正确。 - 使用中文回答。 - 生成相关问题。 - 在回答中引用的段落末尾自然地添引用标识,如[^1]。 用户可见层: - 回答结构清晰,逐步解决问题。 - 尽量保证回答真实可靠。 参考引用: - [^1]: 讨论了Spark DataFrameDataset的发布,以及比较它们的必要性和内容。 - [^2]: DatasetSparkSQL中的一种数据结构,强类型,有指定schema,是对DataFrame API的扩展。提供了类型安全和面向对象的编程接口。 - [^3]: Spark SQL引入了SchemaRDD(后来是DataFrameDataset),支持结构化数据。DataFrameDatasetSpark SQL的API,共享相同的执行引擎。 - [^4]: 展示了使用Dataset的代码示例,包括读取数据、过滤和类型转换。 关键点: - DataFrame:是Spark SQL的一个抽象,表示分布式数据集合,具有命名列(schema)。类似于关系数据库中的表或Pandas DataFrame。它是非类型安全的(schema在运行时检查)。 - Dataset:从Spark 1.6开始引入,是DataFrame的扩展。Dataset是强类型的,提供编译时类型安全。它允许使用Scala或Java的case类来定义类型。 区别: 1. **类型安全**: - DataFrame:是非类型安全的。操作在运行时检查类型错误。 - Dataset:是强类型的,编译时检查类型错误。这意味着在编译时就能捕获类型不匹配的错误[^2]。 2. **编程接口**: - DataFrame:提供类似于SQL的声明式API,支持DSL(领域特定语言)。 - Dataset:提供面向对象的编程接口,可以使用lambda函数和函数式编程[^2][^3]。 3. **语言支持**: - DataFrame:在Scala、Java、Python和R中可用。 - Dataset:主要在Scala和Java中可用,因为Python和R是动态类型语言,不支持编译时类型安全。 4. **性能**: - 两者共享相同的优化和执行引擎(Tungsten引擎),因此在性能上相似。但Dataset的强类型特性可能在某些场景下提供更好的优化[^3]。 5. **创建方式**: - DataFrame:可以从RDD、Hive表、数据源等创建。 - Dataset:可以通过将DataFrame转换Dataset使用as[T]方法)或直接从类型化数据创建。 从引用[^2]:Dataset是对DataFrame API的扩展,提供类型安全和面向对象的编程接口。引用[^3]提到两者共享相同的执行引擎。 结构回答: 1. **介绍**:简要介绍DataFrameDataset。 2. **主要区别**:列出关键区别点。 3. **代码示例**:如果可能,展示简单代码(但用户提供的是Scala代码,我可以用类似格式)。 4. **总结**:何时使用哪个。 5. **相关问题**:生成相关问题。 由于用户要求中文回答,我将用中文。 生成相关问题:基于主题,生成2-3个相关问题。 最后,添引用标识。</think>### Apache SparkDataFrameDataset的区别 在Apache Spark中,DataFrameDataset都是用于处理结构化数据的核心抽象,但它们在设计、类型安全性和编程接口上存在关键差异。以下我将基于Spark官方文档和相关资料,逐步解释它们的区别,帮助您清晰理解。核心区别在于:**DataFrameDataset的非类型安全(schema-on-read)版本,而DatasetDataFrame的类型安全(schema-on-write)扩展**,提供编译时类型检查[^2][^3]。 #### 1. **基本定义** - **DataFrame**: - 表示分布式数据集合,以命名列(schema)组织数据,类似于关系数据库中的表或Pandas DataFrame- 非类型安全:schema在运行时检查,操作错误(如类型不匹配)在运行时抛出异常。 - 支持多语言:Scala、Java、Python和R[^3]。 - 示例:从CSV读取数据时,schema在读取时推断。 ```scala val df = spark.read.option("header", true).csv("data.csv") // schema在运行时确定 ``` - **Dataset**: -DataFrame的强类型扩展,要求显式定义数据类型(例如使用Scala的case类)。 - 类型安全:schema在编译时检查,操作错误(如字段类型错误)在编译期捕获。 - 主要支持静态类型语言:Scala和Java(Python和R因动态类型限制,支持有限)[^2]。 - 示例:定义case类后,DataFrame转换Dataset。 ```scala case class PMData(pm: Double, year: Int) val ds = df.as[PMData] // 编译时类型检查 ``` #### 2. **关键区别对比** 下表总结主要差异(基于Spark 2.0+版本): | 特性 | DataFrame | Dataset | |---------------------|----------------------------------------|----------------------------------------| | **类型安全** | 非类型安全(运行时检查schema) | 强类型安全(编译时检查schema)[^2] | | **编程接口** | 声明式API(类似SQLDSL) | 面向对象API(支持lambda和函数式编程)[^2] | | **语言支持** | Scala、Java、Python、R | 主要Scala和Java | | **性能优化** | 共享Tungsten引擎,优化执行计划相同 | 相同,但强类型可能减少运行时开销[^3] | | **错误处理** | 运行时异常(如ClassCastException) | 编译时错误(如类型不匹配) | | **创建方式** | 从RDD、文件或Hive表创建 | 从DataFrame转换(`as[T]`)或直接创建 | #### 3. **详细解释核心区别** - **类型安全(Type Safety)**: - DataFrame是非类型安全的,因为它使用`Row`对象表示数据,字段类型在运行时解析。例如,在过滤操作中,如果列名错误或类型不匹配,错误只在执行时暴露[^2]。 ```scala df.filter("year > 2015") // 运行时检查"year"是否存在 ``` - Dataset是强类型的,每个元素是特定类型(如自定义case类)。编译时就能捕获错误,提高代码可靠性。引用[^4]中的示例展示了类型转换: ```scala val ds = sourceDF.select( when('PM_Dongsi === "NA", Double.NaN) // 编译时类型检查 .otherwise('PM_Dongsi.cast(DoubleType)) .as("pm") ).as[PMData] // 转换Dataset,编译时验证schema ``` 这减少了生产环境中的bug风险[^2][^4]。 - **编程模型**: - DataFrame使用声明式API,适合SQL-like操作(如`select`、`filter`),便于数据工程师处理ETL任务。 - Dataset提供面向对象风格,支持方法链和lambda表达式,更适合开发复杂业务逻辑的应用程序(如使用`map`或`reduce`)[^2][^3]。 ```scala // Dataset示例:类型安全的操作 ds.filter(_.pm > 50).map(_.year) // 编译时检查pm和year类型 ``` - **性能考虑**: - 两者底层共享相同的优化引擎(Catalyst优化器和Tungsten执行引擎),因此性能差异不大。但在大型数据集上,Dataset的强类型特性可能减少序列化/反序列化开销,因为Encoder直接处理类型化数据[^3]。 - 实际测试中,Dataset在复杂转换(如用户自定义函数)中可能更快,但DataFrame在简单查询中更高效[^2]。 #### 4. **何时使用哪个?** - **使用DataFrame当**: - 需要快速数据探索或SQL-like查询。 - 跨语言工作(如Python或R)。 - 不需要编译时类型检查。 - **使用Dataset当**: - 开发Scala/Java应用,需要高可靠性和类型安全。 - 处理复杂业务逻辑,涉及自定义对象和函数式操作。 - 优化运行时性能(减少类型错误导致的失败)[^2][^3]。 总结:DataFrameDatasetSpark中是互补的。Dataset本质上是DataFrame的类型安全封装,两者在大多数场景下可互换,但Dataset的强类型特性使其更适合企业级应用开发。Spark官方推荐:**优先使用DataFrame进行通用数据处理,在需要类型安全时切换到Dataset**[^2][^3]。 #### 相关问题 1. 如何在Spark中将DataFrame转换Dataset? 2. Dataset的强类型特性如何影响Spark作业的性能? 3. 在Python中是否支持Dataset?如果不支持,有哪些替代方案? [^1]: Recently, there are two new data abstractions released dataframe and datasets in apache spark... [^2]: Spark性能优化 -- > Spark SQLDataFrameDatasetDatasetSparkSQL中的一种数据结构,它是强类型的,包含指定的schema... [^3]: [Spark进阶]-- Spark Dataframe操作。Spark SQL is a component on top of Spark Core that introduces a new data abstraction called SchemaRDD... [^4]: 部分数据展示... 将PM_Dongsi的的数据类型替换成double类型,将这一列的空值替换成double下的nan...
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值