如何旋转 Spark 数据框表？ [复制]

Question

我有这个 3 列的表：

+---+----+----+
| id|type| val|
+---+----+----+
|  1|   A|   0|
|  2|   A|   0|
|  4|   A|   0|
|  2|   B|   1|
|  4|   B|   1|
+---+----+----+

我想把它改成类似：

+---+----+----+
|   |   A|   B|
+---+----+----+
|  1|   0|   -|
|  2|   1|   1|
|  4|   0|   1|
+---+----+----+

我试过了，但没有用：

val data_array = data.pivot(cols=['type'],rows=['id'],values='val')

Answer 1

df.groupBy("id").pivot("type").agg(first("value")).na.fill("-").show

df 是从测试数据文件创建的数据框