Numpy Split Array Based On Condition Notes