Python Program Using Numpy And Function