在 Spark Sql 中透视多列和多行

如何解决在 Spark Sql 中透视多列和多行

我有以下数据。

enter image description here

我必须在 Spark Sql 中旋转才能得到以下结果

enter image description here

尝试了很多方法,在 AWS Glue 中执行起来并不复杂。 任何建议都会有所帮助。

以下是尝试过的,

(select * from category_data
unpivot
(
categoreval
for a in (e1,e2)
));```

解决方法

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

object ColumnToRowDataframe extends App {

  val spark = SparkSession.builder
    .master("local[*]")
    .appName("Task_1")
    .getOrCreate()

  import spark.implicits._

  val df1 = Seq(
    ("Michel","01-08-1999","NY"),("Joel","02-08-1999","WT")
  ).toDF("Name","Dob","Place")

  df1.show(false)
  //  +------+----------+-----+
  //  |Name  |Dob       |Place|
  //  +------+----------+-----+
  //  |Michel|01-08-1999|NY   |
  //  |Joel  |02-08-1999|WT   |
  //  +------+----------+-----+

  val resDF = TransposeDF(df1,Seq("Dob","Place"),"Name")
    .withColumnRenamed("Name","Val")

  resDF.show(false)
  //  +-----+----------+----------+
  //  |Val  |Joel      |Michel    |
  //  +-----+----------+----------+
  //  |Place|WT        |NY        |
  //  |Dob  |02-08-1999|01-08-1999|
  //  +-----+----------+----------+

  // https://nikhil-suthar-bigdata.medium.com/how-to-transpose-spark-dataframe-fa82c079a6b
  def TransposeDF(df: DataFrame,columns: Seq[String],pivotCol: String): DataFrame = {
    val columnsValue = columns.map(x => "'" + x + "'," + x)
    val stackCols = columnsValue.mkString(",")
    val df_1 = df.selectExpr(pivotCol,"stack(" + columns.size + "," + stackCols + ")")
      .select(pivotCol,"col0","col1")

    val final_df = df_1.groupBy(col("col0")).pivot(pivotCol).agg(concat_ws("",collect_list(col("col1"))))
      .withColumnRenamed("col0",pivotCol)
    final_df
  }
}

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?
Java在半透明框架/面板/组件上重新绘画。
Java“ Class.forName()”和“ Class.forName()。newInstance()”之间有什么区别?
在此环境中不提供编译器。也许是在JRE而不是JDK上运行?
Java用相同的方法在一个类中实现两个接口。哪种接口方法被覆盖?
Java 什么是Runtime.getRuntime()。totalMemory()和freeMemory()?
java.library.path中的java.lang.UnsatisfiedLinkError否*****。dll
JavaFX“位置是必需的。” 即使在同一包装中
Java 导入两个具有相同名称的类。怎么处理?
Java 是否应该在HttpServletResponse.getOutputStream()/。getWriter()上调用.close()?
Java RegEx元字符(。)和普通点?