Numpy Apply Function To Array Python W3schools