easier way to split the datasets

[YameiW]

trafficstars

Hello, I rewrite the code to subset the full dataset and split the dataset. I think the logic in my code is easier to understand. I hope it is helpful.

[ds-manav]

can you help me in understanding this code?