Instead of converting to list, save it as numpy arrays