Numpy Map Function On Array In Python