How to use trigonometric functions in Numpy