Top Two Amounts Per Category · Music plays 5

Music plays DataFrame, variant 5. Return rows in the top two distinct amount ranks per category as category, id, amount; include ties. Implement `solve(df)` and return a PySpark DataFrame.