DataFrame的创建&操作


Spark SQL是Spark中的一个模块,主要用于进行结构化数据的处理。它提供的最核心的编程抽象,就是DataFrame。同时Spark SQL还可以作为分布式的SQL查询引擎。Spark SQL最重要的功能之一,就是从Hive中查询数据。

DataFrame是以列的形式组织的,分布式的结构化数据集合。它其实和关系型数据库中的表非常类似,但是底层做了很多的优化。DataFrame可以通过很多来源进行构建,包括:结构化的数据文件,Hive中的表,外部的关系型数据库,以及RDD。有了DataFrame后,可以非常方便地使用DataFrame API和SQL来处理数据。

  • 创建DataFrame(三种)
    • 第一种
    val dfUsers = spark.read.format("csv").option("header", "true").load("file:///root/data/user.csv")
    
    dfUsers.printSchema
    
    // root
    // |-- user_id: string (nullable = true)
    // |-- locale: string (nullable = true)
    // |-- birthyear: string (nullable = true)
    // |-- gender: string (nullable = true)
    // |-- joinedAt: string (nullable = true)
    // |-- location: string (nullable = true)
    // |-- timezone: string (nullable = true)
    
    
    • 第二种
    scala> import org.apache.spark.sql.types._
    import org.apache.spark.sql.types._
    
    scala> import org.apache.spark.sql.Row
    import org.apache.spark.sql.Row
    
    scala> import spark.implicits._
    import spark.implicits._
    
    // 读取文件并转换成RDD[Row]类型
    scala> val uRdd = spark.sparkContext.textFile("file:///root/data/user.csv")
                         .map(x = x.split(","))
                         .mapPartitionsWithIndex((index, iter) => if (index == 0) iter.drop(1) else iter)
                         .map(Row.fromSeq(_))
    uRdd: org.apache.spark.rdd.RDD[org.apache.spark.sql.Row] = MapPartitionsRDD[26]  at map at <console>:30
    
    // 定义Schema
    scala> val schema = StructType(Array(StructField("user_id", StringType, true),       
     StructField("locale", StringType, true),StructField("birthyear", StringType, true),   
     StructField("gender",StringType, true), StructField("joinedAt", StringType, true), 
     StructField("location", StringType, true), StructField("timezone", StringType, true)))
    schema: org.apache.spark.sql.types.StructType = StructType(StructField(user_id,StringType,true), StructField(locale,StringType,true), StructField(birthyear,StringType,true), StructField(gender,StringType,true), StructField(joinedAt,StringType,true), StructField(location,StringType,true), StructField(timezone,StringType,true))
    
    // 创建DataFrame
    scala> val dfUsers = spark.createDataFrame(uRdd, schema)
    dfUsers: org.apache.spark.sql.DataFrame = [user_id: string, locale: string ... 5 more fields]
    
    scala> dfUsers.printSchema
    
    // root
    // |-- user_id: string (nullable = true)
    // |-- locale: string (nullable = true)
    // |-- birthyear: string (nullable = true)
    // |-- gender: string (nullable = true)
    // |-- joinedAt: string (nullable = true)
    // |-- location: string (nullable = true)
    // |-- timezone: string (nullable = true)
    
    scala> dfUsers show 3
    // +----------+------+---------+------+--------------------+------------------+--------+
    // |   user_id|locale|birthyear|gender|            joinedAt|          location|timezone|
    // +----------+------+---------+------+--------------------+------------------+--------+
    // |3197468391| id_ID|     1993|  male|2012-10-02T06:40:...|  Medan  Indonesia|     480|
    // |3537982273| id_ID|     1992|  male|2012-09-29T18:03:...|  Medan  Indonesia|     420|
    // | 823183725| en_US|     1975|  male|2012-10-06T03:14:...|Stratford  Ontario|    -240|
    // +----------+------+---------+------+--------------------+------------------+--------+
    
    
    注:由于该文件首行是列名,所以使用mapPartitionsWithIndex()函数过滤掉
    • 第三种
    scala> val dfUsers = spark.sparkContext.textFile("file:///root/data/users.csv")
                   .map(_.split(","))
                   .mapPartitionsWithIndex((index, iter) => if (index == 0) iter.drop(1) else iter)
                   .map(x => (x(0), x(1), x(2), x(3), x(4), x(5), x(6)))
                   .toDF("user_id", "locale", "birthyear", "gender", "joinedAt", "location", "timezone")
    dfUsers: org.apache.spark.sql.DataFrame = [user_id: string, locale: string ... 5 more  fields]
    
    scala> dfUsers show 3
    // +----------+------+---------+------+--------------------+------------------+--------+
    // |   user_id|locale|birthyear|gender|            joinedAt|          location|timezone|
    // +----------+------+---------+------+--------------------+------------------+--------+
    // |3197468391| id_ID|     1993|  male|2012-10-02T06:40:...|  Medan  Indonesia|     480|
    // |3537982273| id_ID|     1992|  male|2012-09-29T18:03:...|  Medan  Indonesia|     420|
    // | 823183725| en_US|     1975|  male|2012-10-06T03:14:...|Stratford  Ontario|    -240|
    // +----------+------+---------+------+--------------------+------------------+--------+
    
    
    • 把DataFrame持久化到内存
      dfUsers.persist()
      
    • 保存到Hive表
      dfUser.write.mode("overwrite").saveAsTable("sparkdf.users")
      

  • DataFrame Operators

    • 常用操作
    map, flatMap 
    sample, filter 
    sort 
    pipe 
    groupBy, groupByKey, cogroup 
    reduce, reduceByKey, fold 
    partitionBy ▫ zip, union 
    join, crossJoin, leftOuterJoin, rightOuterJoin 
    count, save 
    first, take
    
    • 基本操作
    1、 cache()同步数据的内存
    2、 columns 返回一个string类型的数组,返回值是所有列的名字
    3、 dtypes返回一个string类型的二维数组,返回值是所有列的名字以及类型
    4、 explan()打印执行计划  物理的
    5、 explain(n:Boolean) 输入值为 false 或者true ,返回值是unit  默认是false ,如果输入true 将会打印 逻辑的和物理的
    6、 isLocal 返回值是Boolean类型,如果允许模式是local返回true 否则返回false
    7、 persist(newlevel:StorageLevel) 返回一个dataframe.this.type 输入存储模型类型
    8、 printSchema() 打印出字段名称和类型 按照树状结构来打印
    9、 registerTempTable(tablename:String) 返回Unit ,将df的对象只放在一张表里面,这  个表随着对象的删除而删除了
    10、 schema 返回structType 类型,将字段名称和类型按照结构体类型返回
    11、 toDF()返回一个新的dataframe类型的
    12、 toDF(colnames:String*)将参数中的几个字段返回一个新的dataframe类型的,
    13、 unpersist() 返回dataframe.this.type 类型,去除模式中的数据
    14、 unpersist(blocking:Boolean)返回dataframe.this.type类型 true 和unpersist是一样的作用false 是去除RDD
    
    • 聚合查询操作
    1、 agg(expers:column*) 返回dataframe类型 ,同数学计算求值
          df.agg(max("age"), avg("salary"))
          df.groupBy().agg(max("age"), avg("salary"))
    2、 agg(exprs: Map[String, String])  返回dataframe类型 ,同数学计算求值 map类型的
        df.agg(Map("age" -> "max", "salary" -> "avg"))
        df.groupBy().agg(Map("age" -> "max", "salary" -> "avg"))
    3、 agg(aggExpr: (String, String), aggExprs: (String, String)*)  返回dataframe类型 ,同数学计算求值
        df.agg(Map("age" -> "max", "salary" -> "avg"))
        df.groupBy().agg(Map("age" -> "max", "salary" -> "avg"))
    4、 apply(colName: String) 返回column类型,捕获输入进去列的对象
    5、 as(alias: String) 返回一个新的dataframe类型,就是原来的一个别名
    6、 col(colName: String)  返回column类型,捕获输入进去列的对象
    7、 cube(col1: String, cols: String*) 返回一个GroupedData类型,根据某些字段来汇总
    8、 distinct 去重 返回一个dataframe类型
    9、 drop(col: Column) 删除某列 返回dataframe类型
    10、 dropDuplicates(colNames: Array[String]) 删除相同的列 返回一个dataframe
    11、 except(other: DataFrame) 返回一个dataframe,返回在当前集合存在的在其他集合不存在的
    12、 explode[A, B](inputColumn: String, outputColumn: String)(f: (A) ⇒ TraversableOnce[B])(implicit arg0: scala.reflect.api.JavaUniverse.TypeTag[B]) 返回值是dataframe类型,这个 将一个字段进行更多行的拆分
        df.explode("name","names") {name :String=> name.split(" ")}.show();
        将name字段根据空格来拆分,拆分的字段放在names里面
    13、 filter(conditionExpr: String): 刷选部分数据,返回dataframe类型 df.filter("age>10").show();  df.filter(df("age")>10).show();   df.where(df("age")>10).show(); 都可以
    14、 groupBy(col1: String, cols: String*) 根据某写字段来汇总返回groupedate类型   df.groupBy("age").agg(Map("age" ->"count")).show();df.groupBy("age").avg().show();都可以
    15、 intersect(other: DataFrame) 返回一个dataframe,在2个dataframe都存在的元素
    16、 join(right: DataFrame, joinExprs: Column, joinType: String)
      一个是关联的dataframe,第二个关联的条件,第三个关联的类型:inner, outer, left_outer, right_outer, leftsemi
        df.join(ds,df("name")===ds("name") and  df("age")===ds("age"),"outer").show();
    17、 limit(n: Int) 返回dataframe类型  去n 条数据出来
    18、 na: DataFrameNaFunctions ,可以调用dataframenafunctions的功能区做过滤 df.na.drop().show(); 删除为空的行
    19、 orderBy(sortExprs: Column*) 做alise排序
    20、 select(cols:string*) dataframe 做字段的刷选 df.select($"colA", $"colB" + 1)
    21、 selectExpr(exprs: String*) 做字段的刷选 df.selectExpr("name","name as names","upper(name)","age+1").show();
    22、 sort(sortExprs: Column*) 排序 df.sort(df("age").desc).show(); 默认是asc
    23、 unionAll(other:Dataframe) 合并 df.unionAll(ds).show();
    24、 withColumnRenamed(existingName: String, newName: String) 修改列表 df.withColumnRenamed("name","names").show();
    25、 withColumn(colName: String, col: Column) 增加一列 df.withColumn("aa",df("name")).show();
    
    • 行动操作
    1、 collect() ,返回值是一个数组,返回dataframe集合所有的行
    2、 collectAsList() 返回值是一个java类型的数组,返回dataframe集合所有的行
    3、 count() 返回一个number类型的,返回dataframe集合的行数
    4、 describe(cols: String*) 返回一个通过数学计算的类表值(count, mean, stddev, min, and max),这个可以传多个参数,中间用逗号分隔,如果有字段为空,那么不参与运算,只这对数值类型的字段。例如df.describe("age", "height").show()
    5、 first() 返回第一行 ,类型是row类型
    6、 head() 返回第一行 ,类型是row类型
    7、 head(n:Int)返回n行  ,类型是row 类型
    8、 show()返回dataframe集合的值 默认是20行,返回类型是unit
    9、 show(n:Int)返回n行,,返回值类型是unit
    10、 table(n:Int) 返回n行  ,类型是row 类型
    
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 196,647评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,690评论 2 374
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,739评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,692评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,552评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,410评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,819评论 3 387
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,463评论 0 255
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,752评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,789评论 2 314
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,572评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,414评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,833评论 3 300
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,054评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,345评论 1 254
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,810评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,016评论 2 337

推荐阅读更多精彩内容