partition_gen_by_p function

Create a partition matrix with a partition vector p