SciPy with Matplotlib for visualization

Introduction

SciPy helps us do math and science calculations easily. Matplotlib lets us draw pictures of data so we can understand it better.

You want to find the best curve that fits your data points.

You need to solve math problems like integration or optimization and show the results.

You want to compare two sets of data visually after doing calculations.

You have scientific data and want to plot graphs to explain it clearly.

Syntax

SciPy

import numpy as np
from scipy import optimize
import matplotlib.pyplot as plt

# Define a function to fit
def f(x, a, b):
    return a * x + b

# Sample data
xdata = np.array([1, 2, 3, 4, 5])
ydata = np.array([2, 4, 6, 8, 10])

# Use optimize.curve_fit to find best a, b
params, covariance = optimize.curve_fit(f, xdata, ydata)

# Plot original data and fitted curve
plt.scatter(xdata, ydata)
plt.plot(xdata, f(xdata, *params))
plt.show()

Use optimize.curve_fit to fit a function to data points.

Matplotlib's plt.scatter shows data points, plt.plot draws lines.

Examples

This example fits a straight line to points and shows both on a graph.

SciPy

import numpy as np
from scipy import optimize
import matplotlib.pyplot as plt

# Sample data
xdata = np.array([1, 2, 3, 4, 5])
ydata = np.array([2.1, 4.1, 6.0, 8.1, 10.2])

# Define linear function
def linear(x, a, b):
    return a * x + b

# Fit data
params, _ = optimize.curve_fit(linear, xdata, ydata)

# Plot
plt.scatter(xdata, ydata, label='Data')
plt.plot(xdata, linear(xdata, *params), label='Fit', color='red')
plt.legend()
plt.show()

This example calculates the area under sin(x) from 0 to π and shows it shaded on the plot.

SciPy

import numpy as np
from scipy import integrate
import matplotlib.pyplot as plt

# Define function to integrate
def f(x):
    return np.sin(x)

# Calculate integral from 0 to pi
result, error = integrate.quad(f, 0, np.pi)

# Plot function
x = np.linspace(0, np.pi, 100)
y = f(x)
plt.plot(x, y, label='sin(x)')
plt.fill_between(x, 0, y, alpha=0.3)
plt.title(f'Integral from 0 to pi = {result:.2f}')
plt.legend()
plt.show()

Sample Program

This program creates noisy data points that roughly follow a line. It uses SciPy to find the best line that fits the points. Then it draws the points and the fitted line on a graph.

SciPy

import numpy as np
from scipy import optimize
import matplotlib.pyplot as plt

# Data points with some noise
xdata = np.linspace(0, 10, 20)
ydata = 3.5 * xdata + 2 + np.random.normal(0, 2, size=xdata.size)

# Define linear function to fit
def linear_func(x, a, b):
    return a * x + b

# Fit the data
params, covariance = optimize.curve_fit(linear_func, xdata, ydata)

# Print fitted parameters
a_fit, b_fit = params
print(f'Fitted line: y = {a_fit:.2f} * x + {b_fit:.2f}')

# Plot data and fitted line
plt.scatter(xdata, ydata, label='Data points')
plt.plot(xdata, linear_func(xdata, *params), color='red', label='Fitted line')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Linear Fit using SciPy and Matplotlib')
plt.legend()
plt.show()

Important Notes

Always check your data before fitting to avoid errors.

Matplotlib plots appear in a window or inline if using notebooks.

Fitting results depend on data quality and initial guesses.

Summary

SciPy helps with math tasks like fitting and integration.

Matplotlib shows data and results visually with plots.

Combining both makes data science easier and clearer.

Practice

1. What is the main purpose of using SciPy together with Matplotlib in data science?

easy

A. To write text documents automatically

B. To create websites with interactive buttons

C. To store large amounts of data in databases

D. To perform mathematical calculations and then visualize the results

Solution

1. Step 1: Understand SciPy's role

   SciPy is used for math tasks like integration, optimization, and fitting data.

2. Step 2: Understand Matplotlib's role

   Matplotlib is used to create visual plots to show data and results clearly.

3. Final Answer:

   To perform mathematical calculations and then visualize the results -> Option D

4. Quick Check:

   SciPy + Matplotlib = Math + Visualization [OK]

Hint: SciPy does math, Matplotlib draws pictures [OK]

Common Mistakes:

• Confusing SciPy with web development tools
• Thinking Matplotlib stores data
• Assuming SciPy creates visual plots

2. Which of the following is the correct way to import SciPy's integrate module and Matplotlib's pyplot for plotting?

easy

A. import scipy.integrate as integrate
import matplotlib.pyplot as plt

B. import scipy.plot as sp
import matplotlib as mpl

C. from scipy import plot
import matplotlib.pyplot

D. import scipy.integrate as sp
import matplotlib.pyplot as matplotlib

Solution

1. Step 1: Check SciPy import syntax

   The correct way is to import the integrate module as 'integrate' for clarity.

2. Step 2: Check Matplotlib import syntax

   Matplotlib's pyplot is commonly imported as 'plt' for easy plotting commands.

3. Final Answer:

   import scipy.integrate as integrate import matplotlib.pyplot as plt -> Option A

4. Quick Check:

   Standard imports use 'as integrate' and 'as plt' [OK]

Hint: Use 'as integrate' and 'as plt' for clear code [OK]

Common Mistakes:

• Using wrong module names like scipy.plot
• Not aliasing pyplot as plt
• Importing entire matplotlib instead of pyplot

3. What will the following code display?

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import quad

def f(x):
    return np.sin(x)

result, error = quad(f, 0, np.pi)
plt.plot([0, np.pi], [0, result])
plt.title(f"Integral result: {result:.2f}")
plt.show()

medium

A. A line plot from 0 to π with y-values 0 to approximately 2 showing the integral result

B. A scatter plot of sine values between 0 and π

C. A bar chart showing the error value of the integral

D. An empty plot with no lines or points

Solution

1. Step 1: Understand the integral calculation

   The code calculates the integral of sin(x) from 0 to π, which equals 2.

2. Step 2: Understand the plot command

   It plots a line from (0,0) to (π, result), so from 0 to π on x-axis and 0 to ~2 on y-axis.

3. Final Answer:

   A line plot from 0 to π with y-values 0 to approximately 2 showing the integral result -> Option A

4. Quick Check:

   Integral of sin(x) 0 to π = 2, line plot shows this [OK]

Hint: Integral of sin(x) from 0 to π is 2, plot line shows it [OK]

Common Mistakes:

• Thinking the plot shows sine wave points
• Confusing scatter plot with line plot
• Ignoring the integral result in the plot

4. The following code is intended to plot the cumulative integral of cos(x) from 0 to 2π, but it raises an error. What is the error and how to fix it?

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import cumtrapz

x = np.linspace(0, 2*np.pi, 100)
y = np.cos(x)

integral = cumtrapz(y, x)
plt.plot(x, integral)
plt.show()

medium

A. Error: plt.plot cannot plot arrays; fix by converting to list

B. Error: cumtrapz returns array shorter by 1; fix by plotting plt.plot(x[1:], integral)

C. Error: cumtrapz needs y first then x; fix by swapping arguments

D. Error: np.cos requires integer input; fix by converting x to int

Solution

1. Step 1: Identify cumtrapz output length

   cumtrapz returns an array with length one less than input arrays.

2. Step 2: Fix plotting mismatch

   Plot x[1:] with integral to match array sizes and avoid error.

3. Final Answer:

   Error: cumtrapz returns array shorter by 1; fix by plotting plt.plot(x[1:], integral) -> Option B

4. Quick Check:

   cumtrapz output length = input length - 1 [OK]

Hint: cumtrapz output shorter by 1, plot with x[1:] [OK]

Common Mistakes:

• Plotting full x with shorter integral array
• Trying to convert floats to int unnecessarily
• Swapping arguments of cumtrapz incorrectly

5. You want to fit a Gaussian curve to noisy data points and then plot both the data and the fitted curve. Which approach correctly uses SciPy and Matplotlib together?

hard

A. Use matplotlib.pyplot.scatter to fit the curve, then plot with scipy.optimize.curve_fit

B. Use scipy.integrate.quad to fit the curve, then plot with matplotlib.pyplot.bar

C. Use scipy.optimize.curve_fit to find parameters, then plot data points and fitted curve with matplotlib.pyplot

D. Use numpy.polyfit to fit the curve, then plot with scipy.integrate.cumtrapz

Solution

1. Step 1: Choose fitting method

   scipy.optimize.curve_fit is designed to fit functions like Gaussian to data.

2. Step 2: Plot data and fit

   Use matplotlib.pyplot to plot original data points and the smooth fitted curve.

3. Final Answer:

   Use scipy.optimize.curve_fit to find parameters, then plot data points and fitted curve with matplotlib.pyplot -> Option C

4. Quick Check:

   curve_fit fits, pyplot plots data and fit [OK]

Hint: curve_fit fits data, pyplot plots results [OK]

Common Mistakes:

• Using integration functions for fitting
• Mixing plotting and fitting functions incorrectly
• Using bar plots for continuous curve visualization