https://remp3indir.net/numpy-array-hackerrank-solution/