Pyspark: rearranging pyspark nested schema dynamically

Question

I have a pypspark schema that looks like this

root
|--field1
|--field2
| |--element: struct
| | |--field01
| | |--field02

I need to recreate