Finding the root of a mathematical function
In this short recipe, we will see how to use SciPy to find the root of a simple mathematical function of a single real variable.
How to do it...
- Let's import NumPy, SciPy,
scipy.optimize
, and matplotlib:>>> import numpy as np import scipy as sp import scipy.optimize as opt import matplotlib.pyplot as plt %matplotlib inline
-
We define the mathematical function in Python. We will try to find a root of this function numerically. Here, a root corresponds to a fixed point of the cosine function:
>>> def f(x): return np.cos(x) - x
-
Let's plot this function on the interval (using 1000 samples):
>>> x = np.linspace(-5, ...
Get IPython Interactive Computing and Visualization Cookbook - Second Edition now with the O’Reilly learning platform.
O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.