Train Validation Test Split Python String Into Array