按列值将数据框拆分为多个数据框,使用Scala编写。
创始人
2024-11-03 06:00:31
0

以下是一个示例代码,它将数据框按列值拆分为多个数据框:

import org.apache.spark.sql.{DataFrame, SparkSession}

object DataFrameSplitter {
  def splitDataFrameByColumnValue(df: DataFrame, column: String): Map[String, DataFrame] = {
    val distinctValues = df.select(column).distinct().collect()

    distinctValues.map(row => {
      val value = row.get(0).toString
      val filteredDF = df.filter(df(column) === value)
      value -> filteredDF
    }).toMap
  }

  def main(args: Array[String]): Unit = {
    val spark = SparkSession.builder()
      .appName("DataFrameSplitter")
      .master("local")
      .getOrCreate()

    import spark.implicits._

    val data = Seq(
      ("Alice", 25, "New York"),
      ("Bob", 30, "London"),
      ("Charlie", 35, "New York"),
      ("David", 40, "London"),
      ("Eve", 45, "New York")
    )

    val df = data.toDF("Name", "Age", "City")

    val splitDataFrames = splitDataFrameByColumnValue(df, "City")

    splitDataFrames.foreach { case (value, df) =>
      println(s"Data frame with $value:")
      df.show()
    }

    spark.stop()
  }
}

此示例代码使用SparkSession创建一个Spark应用程序。它基于给定的列名将数据框拆分为多个数据框。您可以根据需要更改数据源和要拆分的列名。

在示例代码中,我们使用了一个名为splitDataFrameByColumnValue的函数来执行拆分操作。此函数首先使用distinct()方法获取指定列的唯一值。然后,它使用filter()方法将数据框按列值筛选,并将每个筛选后的数据框与列值关联为键值对。最后,它将这些键值对映射到一个Map中,并将其作为结果返回。

main()方法中,我们创建了一个包含一些示例数据的数据框。然后,我们调用splitDataFrameByColumnValue()函数将数据框按“City”列值拆分。最后,我们遍历拆分后的数据框并打印结果。

相关内容

热门资讯

事发当天!上饶辅助设备出租,真... 事发当天!上饶辅助设备出租,真是是真的辅助工具(有挂头条)-哔哩哔哩上饶辅助设备出租脚本下载中分为三...
随着!菠萝神辅助器app,一直... 随着!菠萝神辅助器app,一直是有辅助平台(有挂教程)-哔哩哔哩1、很好的工具软件,可以解锁游戏的菠...
推出新举措!兴动海满辅助,一贯... 您好,兴动海满辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】很多玩家在...
做出回应!微乐贵阳捉鸡麻将挂软... 做出回应!微乐贵阳捉鸡麻将挂软件,果然真的是有辅助神器(有挂秘诀)-哔哩哔哩1、打开软件启动之后找到...
反观!九游破解辅助插件,都是真... 反观!九游破解辅助插件,都是真的有辅助神器(有挂秘籍)-哔哩哔哩九游破解辅助插件能透视中分为三种模型...
据文件显示!科米台州麻将辅助,... 据文件显示!科米台州麻将辅助,真是有挂辅助app(竟然有挂)-哔哩哔哩1、每一步都需要思考,不同水平...
今天上午!赣湘互娱挂,总是是有... 今天上午!赣湘互娱挂,总是是有辅助修改器(有挂详情)-哔哩哔哩1、在赣湘互娱挂插件功能辅助器技巧中,...
来临!广西友乐解码器辅助器,原... 来临!广西友乐解码器辅助器,原来真的是有辅助脚本(真的有挂)-哔哩哔哩1.广西友乐解码器辅助器 选牌...
来临!新天道辅助脚本,确实有挂... 来临!新天道辅助脚本,确实有挂辅助下载(有挂秘诀)-哔哩哔哩1、金币登录送、破产送、升级送、活动送。...
近期!青橙竞技卡五星辅助,好像... 近期!青橙竞技卡五星辅助,好像真的是有辅助脚本(竟然有挂)-哔哩哔哩1、用户打开应用后不用登录就可以...