How to split the dataset into train and test based on client number using “Federated learning”

Question

I try to use np.array.split to split the dataset into 2 part, but it does not work well

Hope some one can give some advice on this issue

x` (i