Matlab Split Dataset In Python