Separate data in four groups, and put each group in separate list- Python

Question

I have a dataset(.csv file), having four different classes(0,1,2,3). I have already used the below function to group the data:

data = pd.read_csv("train_or