指点成金-最美分享吧

登录

spark创建空dataframe

佚名 举报

篇首语:本文由小编为大家整理,主要介绍了spark创建空dataframe相关的知识,希望对你有一定的参考价值。

1.emptyDataFrame

项目中会需要用到创建空的dataframe。spark提供了emptyDataFrame方法,可以直接创建。

  /**   * Returns a `DataFrame` with no rows or columns.   *   * @since 2.0.0   */  @transient  lazy val emptyDataFrame: DataFrame =     createDataFrame(sparkContext.emptyRDD[Row], StructType(Nil))  

源码中已经很明确告诉我们,emptyDataFrame创建了一个不含任何行列且schema为空的dataframe。

这个空的dataframe在实际中用途有限,比如如下场景。

def t2() = val sparkConf = new SparkConf().setMaster("local[2]")val spark = SparkSession.builder().config(sparkConf).getOrCreate()import spark.implicits._val seq = Seq(("id1", 1), ("id1", 1), ("id1", 2), ("id2", 1), ("id2", 1))val df = spark.sparkContext.parallelize(seq, 1).toDF("id", "num")var emptydf = spark.emptyDataFrameemptydf = emptydf.union(df)emptydf.show(10)

上面的方法运行时候直接抛出错误信息

org.apache.spark.sql.AnalysisException: Union can only be performed on tables with the same number of columns, but the first table has 0 columns and the second table has 2 columns;;"Union:- LogicalRDD+- Project [_1#3 AS id#6, _2#4 AS num#7]   +- SerializeFromObject [staticinvoke(class org.apache.spark.unsafe.types.UTF8String, StringType, fromString, assertnotnull(input[0, scala.Tuple2, true], top level Product input object)._1, true) AS _1#3, assertnotnull(input[0, scala.Tuple2, true], top level Product input object)._2 AS _2#4]      +- ExternalRDD [obj#2]

错误信息也很直观,emptydf是0 columns,df是2 columns,所以无法直接union。

2.emptyDataFrame指定schema

为了解决上面的问题,我们需要在生成emptyDataFrame的时候指定schema。

import org.apache.spark.sql.types.IntegerType, StringType, StructField, StructTypedef t2() = val sparkConf = new SparkConf().setMaster("local[2]")val spark = SparkSession.builder().config(sparkConf).getOrCreate()import spark.implicits._val seq = Seq(("id1", 1), ("id1", 1), ("id1", 2), ("id2", 1), ("id2", 1))val df = spark.sparkContext.parallelize(seq, 1).toDF("id", "num")val schema = StructType(Seq(StructField("id", StringType, true),StructField("num", IntegerType, true)))var emptydf = spark.createDataFrame(spark.sparkContext.emptyRDD[Row], schema)emptydf = emptydf.union(df)emptydf.show(10)

通过StructType与StructField配合使用,就可以实现上述目的。

以上是关于spark创建空dataframe的主要内容,如果未能解决你的问题,请参考以下文章