Replace Numpy Array Elements Javascript Interview