進(jìn)一步理解DataFrame, Dataset, RDD

Dataset=RDD+schema

Dataset幾乎就是一個(gè)RDD,除了它還包括一個(gè)schema吕座,這個(gè)schema很多時(shí)候也是自動(dòng)推導(dǎo)出來的,最簡(jiǎn)單的schema是包含一個(gè)名為value的列,它的類型可以是String弦追,Int...

如下代碼創(chuàng)建一個(gè)Dataset:

scala> import spark.implicits._
import spark.implicits._

scala> val ds = Seq(("bluejoe", 100), ("alex", 200)).toDS
ds: org.apache.spark.sql.Dataset[(String, Int)] = [_1: string, _2: int]

scala> ds.schema
res0: org.apache.spark.sql.types.StructType = StructType(StructField(_1,StringType,true), StructField(_2,IntegerType,false))

scala> ds.collect
res1: Array[(String, Int)] = Array((bluejoe,100), (alex,200))

這個(gè)Dataset就包含了2行記錄,每個(gè)記錄是一個(gè)Tuple2花竞,如:(bluejoe,100)

可以針對(duì)這個(gè)Dataset做SQL查詢:

scala> ds.select("_1").collect
res4: Array[org.apache.spark.sql.Row] = Array([bluejoe], [alex])

scala> ds.show
+-------+---+
|     _1| _2|
+-------+---+
|bluejoe|100|
|   alex|200|
+-------+---+
scala> ds.select("_1").show
+-------+
|     _1|
+-------+
|bluejoe|
|   alex|
+-------+
scala> ds.select(ds("_1")).show
+-------+
|     _1|
+-------+
|bluejoe|
|   alex|
+-------+
scala> ds.select($"_1").show
+-------+
|     _1|
+-------+
|bluejoe|
|   alex|
+-------+

ds.select("_1")與ds.select(ds("_1"))劲件,以及ds.select($"_1")等價(jià)
$"_1"神奇嗎?一點(diǎn)都不神奇约急,$()其實(shí)是一個(gè)函數(shù):

  implicit class StringToColumn(val sc: StringContext) {
    def $(args: Any*): ColumnName = {
      new ColumnName(sc.s(args: _*))
    }
  }

SQL列還可以進(jìn)行運(yùn)算操作:

scala> ds.select(ds("_2")+10).show
+---------+
|(_2 + 10)|
+---------+
|      110|
|      210|
+---------+
scala> ds.select($"_2"+10).show
+---------+
|(_2 + 10)|
+---------+
|      110|
|      210|
+---------+

+零远、-等運(yùn)算符其實(shí)也被ColumnName定義了,這里不再贅述厌蔽。

也可以使用map()對(duì)Dataset進(jìn)行變形:

scala> ds.map(x=>(x._1.toUpperCase, x._2+10)).show
+-------+---+
|     _1| _2|
+-------+---+
|BLUEJOE|110|
|   ALEX|210|
+-------+---+

可以看出牵辣,map()函數(shù)會(huì)生成新的schema:

scala> ds.map(x=>(x._1.toUpperCase, x._2+10, true)).show
+-------+---+----+
|     _1| _2|  _3|
+-------+---+----+
|BLUEJOE|110|true|
|   ALEX|210|true|
+-------+---+----+

除了將一個(gè)Tuple轉(zhuǎn)換成另外一個(gè)Tuple,還可以轉(zhuǎn)成一個(gè)JavaBean:

scala> case class Person(name:String,age:Int){};
defined class Person
scala> val ds2=ds.map(x=>Person(x._1.toUpperCase, x._2+10))
ds2: org.apache.spark.sql.Dataset[Person] = [name: string, age: int]

scala> ds2.show
+-------+---+
|   name|age|
+-------+---+
|BLUEJOE|110|
|   ALEX|210|
+-------+---+

注意這個(gè)新的Dataset的每一行變成了一個(gè)Person對(duì)象:

scala> ds2.collect
res36: Array[Person] = Array(Person(BLUEJOE,110), Person(ALEX,210))

注意奴饮,不是任何對(duì)象都可以放到Dataset中:

scala> import org.apache.spark.sql._
import org.apache.spark.sql._

scala> ds.map(x=>Row(x._1.toUpperCase, x._2+10)).show
<console>:32: error: Unable to find encoder for type stored in a Dataset.  Primitive types (Int, String, etc) and Product types (case classes) are supported by importing spark.implicits._  Support for serializing other types will be added in future releases.
       ds.map(x=>Row(x._1.toUpperCase, x._2+10)).show

DataFrame是Dataset[Row]的別名
A DataFrame is a Dataset organized into named columns.

Dataset可以轉(zhuǎn)成DataFrame:

scala> val df=ds.toDF
df: org.apache.spark.sql.DataFrame = [_1: string, _2: int]

scala> df.collect
res33: Array[org.apache.spark.sql.Row] = Array([bluejoe,100], [alex,200])

注意看到DataFrame的每行確實(shí)是一個(gè)Row纬向,觀察源代碼:

 def toDF(): DataFrame = new Dataset[Row](sparkSession, queryExecution, RowEncoder(schema))

實(shí)際上,toDF()使用一個(gè)RowEncoder來實(shí)現(xiàn)Tuple到Row的轉(zhuǎn)碼戴卜。

也可以使用as()函數(shù)來轉(zhuǎn)換成DataFrame:

scala> ds.as[Row](RowEncoder(ds.schema)).collect
res55: Array[org.apache.spark.sql.Row] = Array([bluejoe,100], [alex,200])

DataFrame的map()函數(shù)具有一些陷阱逾条,因?yàn)樗鼘?shí)際上還是一個(gè)Dataset,所以它的每一行還是可以轉(zhuǎn)換成任意對(duì)象(甚至是非Row對(duì)象M栋Jχ):

scala> df.map(x=>(x(0).asInstanceOf[String].toLowerCase, x(1).asInstanceOf[Int]-10)).collect
res43: Array[(String, Int)] = Array((bluejoe,90), (alex,190))

看到?jīng)]?這個(gè)map()之后的對(duì)象并不再是DataFrame了=恰吃警!如果堅(jiān)持要轉(zhuǎn)變成DataFrame,就必須用到別扭的toDF():

scala> df.map(x=>(x(0).asInstanceOf[String].toLowerCase, x(1).asInstanceOf[Int]-10)).toDF.collect
res44: Array[org.apache.spark.sql.Row] = Array([bluejoe,90], [alex,190])

或者指定Encoder:

scala> df.map{x:Row=>Row(x(0).asInstanceOf[String].toLowerCase, x(1).asInstanceOf[Int]-10)}(RowEncoder(ds.schema)).collect
res52: Array[org.apache.spark.sql.Row] = Array([bluejoe,90], [alex,190])

別扭嗎泳桦?真的很別扭汤徽!

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市灸撰,隨后出現(xiàn)的幾起案子谒府,更是在濱河造成了極大的恐慌,老刑警劉巖浮毯,帶你破解...
    沈念sama閱讀 222,946評(píng)論 6 518
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件完疫,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡债蓝,警方通過查閱死者的電腦和手機(jī)壳鹤,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,336評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來饰迹,“玉大人芳誓,你說我怎么就攤上這事余舶。” “怎么了锹淌?”我有些...
    開封第一講書人閱讀 169,716評(píng)論 0 364
  • 文/不壞的土叔 我叫張陵匿值,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我赂摆,道長(zhǎng)挟憔,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 60,222評(píng)論 1 300
  • 正文 為了忘掉前任烟号,我火速辦了婚禮绊谭,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘汪拥。我一直安慰自己达传,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,223評(píng)論 6 398
  • 文/花漫 我一把揭開白布迫筑。 她就那樣靜靜地躺著趟大,像睡著了一般。 火紅的嫁衣襯著肌膚如雪铣焊。 梳的紋絲不亂的頭發(fā)上逊朽,一...
    開封第一講書人閱讀 52,807評(píng)論 1 314
  • 那天,我揣著相機(jī)與錄音曲伊,去河邊找鬼叽讳。 笑死,一個(gè)胖子當(dāng)著我的面吹牛坟募,可吹牛的內(nèi)容都是我干的岛蚤。 我是一名探鬼主播,決...
    沈念sama閱讀 41,235評(píng)論 3 424
  • 文/蒼蘭香墨 我猛地睜開眼懈糯,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼涤妒!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起赚哗,我...
    開封第一講書人閱讀 40,189評(píng)論 0 277
  • 序言:老撾萬榮一對(duì)情侶失蹤她紫,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后屿储,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體贿讹,經(jīng)...
    沈念sama閱讀 46,712評(píng)論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,775評(píng)論 3 343
  • 正文 我和宋清朗相戀三年够掠,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了民褂。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,926評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖赊堪,靈堂內(nèi)的尸體忽然破棺而出面殖,到底是詐尸還是另有隱情,我是刑警寧澤哭廉,帶...
    沈念sama閱讀 36,580評(píng)論 5 351
  • 正文 年R本政府宣布畜普,位于F島的核電站,受9級(jí)特大地震影響群叶,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜钝荡,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,259評(píng)論 3 336
  • 文/蒙蒙 一街立、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧埠通,春花似錦赎离、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,750評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至舞蔽,卻和暖如春荣病,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背渗柿。 一陣腳步聲響...
    開封第一講書人閱讀 33,867評(píng)論 1 274
  • 我被黑心中介騙來泰國打工个盆, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人朵栖。 一個(gè)月前我還...
    沈念sama閱讀 49,368評(píng)論 3 379
  • 正文 我出身青樓颊亮,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國和親陨溅。 傳聞我的和親對(duì)象是個(gè)殘疾皇子终惑,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,930評(píng)論 2 361

推薦閱讀更多精彩內(nèi)容