Root finding (root, brentq) in SciPy

Introduction

Root finding helps us find where a function equals zero. This is useful to solve real problems like finding break-even points or crossing points.

You want to find the exact point where a cost equals revenue.

You need to solve equations that cannot be rearranged easily.

You want to find the zero crossing of a sensor signal.

You want to find the temperature where a reaction rate becomes zero.

You want to find the root of a polynomial or any continuous function.

Syntax

SciPy

from scipy.optimize import root, brentq

# root usage
def func(x):
    return x**2 - 4

sol = root(func, x0=1)  # x0 is initial guess

# brentq usage
def func2(x):
    return x**3 - x - 2

root_value = brentq(func2, 1, 2)  # a and b bracket the root

root needs an initial guess and can solve many types of equations.

brentq needs an interval where the function changes sign (one positive, one negative).

Examples

Find root of x² - 9 starting from guess 2. The root is 3 or -3, root returns one close to guess.

SciPy

from scipy.optimize import root

def f(x):
    return x**2 - 9

solution = root(f, x0=2)
print(solution.x)

Find root of x³ - 1 between 0 and 2. The root is 1.

SciPy

from scipy.optimize import brentq

def f(x):
    return x**3 - 1

root_val = brentq(f, 0, 2)
print(root_val)

Sample Program

This program finds roots of two functions. The first uses root with a guess. The second uses brentq with an interval where the function changes sign.

SciPy

from scipy.optimize import root, brentq

def func1(x):
    return x**2 - 16

def func2(x):
    return x**3 - 2*x - 5

# Using root with initial guess
sol1 = root(func1, x0=1)

# Using brentq with interval
sol2 = brentq(func2, 2, 3)

print(f"Root found by root: {sol1.x[0]:.4f}")
print(f"Root found by brentq: {sol2:.4f}")

Important Notes

Always check if the function changes sign in the interval when using brentq.

root can find complex roots but needs a good initial guess.

Both methods return numerical approximations, not exact answers.

Summary

Root finding helps locate where a function equals zero.

root uses an initial guess and is flexible.

brentq needs an interval with a sign change and is very reliable.

Practice

1. What is the main purpose of the scipy.optimize.root function in root finding?

easy

A. To calculate the maximum value of a function

B. To integrate a function over an interval

C. To find where a function equals zero using an initial guess

D. To sort an array of numbers

Solution

1. Step 1: Understand the purpose of root finding

   Root finding means locating points where a function's value is zero.

2. Step 2: Identify what scipy.optimize.root does

   This function finds roots starting from an initial guess, aiming to solve f(x) = 0.

3. Final Answer:

   To find where a function equals zero using an initial guess -> Option C

4. Quick Check:

   Root finding = find zero points [OK]

Hint: Root means function value zero, root() finds it from guess [OK]

Common Mistakes:

• Confusing root finding with integration
• Thinking root finds maxima or minima
• Assuming root sorts data

2. Which of the following is the correct way to import the brentq function from scipy.optimize?

easy

A. from scipy.optimize import brentq

B. import brentq from scipy.optimize

C. from scipy import brentq

D. import scipy.optimize.brentq

Solution

1. Step 1: Recall Python import syntax for functions

   To import a function from a module, use: from module import function.

2. Step 2: Apply to brentq from scipy.optimize

   The correct syntax is: from scipy.optimize import brentq.

3. Final Answer:

   from scipy.optimize import brentq -> Option A

4. Quick Check:

   Correct import syntax = from module import function [OK]

Hint: Use 'from module import function' to import specific functions [OK]

Common Mistakes:

• Using 'import function from module' which is invalid
• Trying to import directly from scipy without optimize
• Using dot notation in import statement incorrectly

3. What will be the output of this code snippet?

from scipy.optimize import brentq

def f(x):
    return x**3 - 2*x - 5

root = brentq(f, 2, 3)
print(round(root, 3))

medium

A. 2.095

B. 2.0946

C. 2.0945

D. 2.0946 (rounded to 4 decimals)

Solution

1. Step 1: Understand the function and interval

   The function is f(x) = x³ - 2x - 5. We look for root between 2 and 3.

2. Step 2: Use brentq to find root and round output

   Using brentq finds root near 2.0945514815. Rounded to 3 decimals is 2.095.

3. Step 3: Check rounding precision in options

   Options B (2.0946), C (2.0945), D (notes 4 decimals) are near unrounded root, but print(round(root,3)) outputs 2.095.

4. Final Answer:

   2.095 -> Option A

5. Quick Check:

   Root near 2.095 rounded 3 decimals = 2.095 [OK]

Hint: brentq finds root in interval; round final answer carefully [OK]

Common Mistakes:

• Confusing rounding digits
• Using wrong interval causing error
• Misunderstanding brentq output as exact integer

4. Identify the error in this code snippet and choose the correct fix:

from scipy.optimize import brentq

def g(x):
    return x**2 - 4

root = brentq(g, 3, 5)
print(root)

medium

A. No error; code runs fine

B. Interval must include a sign change; use brentq(g, 1, 3)

C. Function g(x) is incorrect; change to g(x) = x**2 + 4

D. Import statement is wrong; use import brentq from scipy.optimize

Solution

1. Step 1: Check function values at interval ends

   g(3) = 9 - 4 = 5 (positive), g(5) = 25 - 4 = 21 (positive). No sign change.

2. Step 2: Understand brentq requires sign change

   brentq needs f(a) and f(b) to have opposite signs to find root.

3. Step 3: Fix interval to include root

   Use interval (1, 3) because g(1) = -3 (negative), g(3) = 5 (positive), sign change exists.

4. Final Answer:

   Interval must include a sign change; use brentq(g, 1, 3) -> Option B

5. Quick Check:

   brentq needs sign change in interval [OK]

Hint: Check function signs at interval ends before brentq [OK]

Common Mistakes:

• Ignoring sign change requirement
• Changing function unnecessarily
• Incorrect import syntax

5. You want to find the root of h(x) = cos(x) - x using scipy.optimize.root. Which initial guess is most appropriate and why?

from scipy.optimize import root
import numpy as np

def h(x):
    return np.cos(x) - x

sol = root(h, ?)
print(sol.x)

hard

A. -1, because negative guesses always work better

B. 2, because cos(2) is near zero

C. 5, because root is always positive and large

D. 0.5, because root is near where cos(x) crosses x close to 0.5

Solution

1. Step 1: Understand the function behavior

   h(x) = cos(x) - x crosses zero near 0.7 (between 0 and 1).

2. Step 2: Choose initial guess close to root

   Initial guess 0.5 is close to root, helping root() converge quickly.

3. Step 3: Evaluate other options

   2 and 5 are far from root; -1 is negative and root is positive, so less suitable.

4. Final Answer:

   0.5, because root is near where cos(x) crosses x close to 0.5 -> Option D

5. Quick Check:

   Good initial guess near root speeds root() convergence [OK]

Hint: Pick initial guess near expected root for root() [OK]

Common Mistakes:

• Choosing initial guess far from root
• Assuming root is always positive and large
• Believing negative guesses always work better