How to do stratified sampling on two columns in PySpark Dataframe?

Question

I want to sample below data set based on IDs and the comm_type they fall into; The same IDs can have multiple comm_types, the data set is huge so I want to do further analysis o