Error zsh: killed when running python program that loads a txt file

Question

So I need to split a txt file into a testing and a training file (also txt). I\'ve run the code below for a smaller data set and it works perfectly. But it fails when I try