Pyspark expand rows

Question

I have a df as below

| Code | Begin_Dt   | End_Dt         | Difference
| ------------------| -------------- |-----------
| AA   | 2021-01-01 | 2021-01-31     |30