group rows (ID) and then assign a treatment for each group ID

Question

I am working with a long (person-period) dataset with unbalanced number (varying N of observations per person). What I want to do is randomly assign a treatment (A, B or C) to e