How To Write Trigonometric Function In Python