python code to compute polynomial equation