How to build chain of segments in scope of pyspark dataframe

Question

I have a huge pyspark dataframe with segments and their subsegments, like this:

SegmentId  SubSegmentStart SubSegmentEnd
   1         a1               a2
   1