分解表以透视列（SQL，PYSPARK）

習慣笑著流淚

我正在AWS Glue中使用python3.6的环境pyspark中进行工作。我有这张桌子：
$ m$ v4 y* h2 D* \3 [+----+-----+-----+-----+
|year|month|total| loop|
+----+-----+-----+-----+
|2012|    1|   20|loop1|
|2012|    2|   30|loop1|
% Y. X2 k& r. q5 o9 E; F|2012|    1|   10|loop2|
|2012|    2|    5|loop2|
|2012|    1|   50|loop3|
; {3 [4 j& K3 h|2012|    2|   60|loop3|
+----+-----+-----+-----+
我需要得到一个类似的输出：
year    month   total_loop1 total_loop2 total_loop3
2 o) u* Z$ S  M% {6 J4 d2012    1         20           10           50
& |9 V- t" {$ U7 o2 T" O; N  h" Z2012    2         30           5            60
我得到的是与SQL代码更接近的信息：
select a.year,a.month, a.total,b.total from test a
3 l' @! j7 Q  ^4 f& Zleft join test b
on a.loop  b.loop
7 E0 [; v9 ?9 Jand a.year = b.year and a.month=b.month
8 ]$ P  A4 Y. F, ^9 U" A) [3 t到目前为止的输出：
+----+-----+-----+-----+
' e' m  ]( Z" Y3 M|year|month|total|total|
/ M; q* c) C8 v: F3 h; |5 V+ ~+----+-----+-----+-----+
1 d- @- o; I8 _|2012|    1|   20|   10|
|2012|    1|   20|   50|
3 |* f3 g# l4 o" f|2012|    1|   10|   20|
6 B7 _6 M0 Y* I+ F1 w+ c1 F|2012|    1|   10|   50|
5 a! G8 j7 h0 h9 }: \|2012|    1|   50|   20|
|2012|    1|   50|   10|
0 o, O( [9 e' ?3 B7 s  o|2012|    2|   30|    5|
|2012|    2|   30|   60|
9 m, S4 L( a, {% p# K* ~|2012|    2|    5|   30|
0 R! {' u/ V7 A+ l1 e! R6 s|2012|    2|    5|   60|
|2012|    2|   60|   30|
) u4 E, ^$ Z- p! ]+ C8 J|2012|    2|   60|    5|
- n7 e& i, Q  M& W5 B+----+-----+-----+-----+
, y/ i. J$ v- f  W我该怎么办？非常感谢
, F8 ?4 p6 R+ t! G+ R" H" L. c               
6 b5 Y& J5 H; M/ K' J! s解决方案: