Numpy Array Manipulation Hackerrank