SciPy with Matplotlib for visualization - Mini Project: Build & Apply

NumPy with Matplotlib for visualization

📖 Scenario: You are a data analyst working with scientific data. You want to understand how a mathematical function behaves and visualize it clearly.

🎯 Goal: Build a Python program that uses NumPy to create data points for a sine wave and then uses Matplotlib to plot this sine wave graph.

📋 What You'll Learn

Create an array of x values using NumPy's linspace function

Calculate the sine of each x value using NumPy's sine function

Plot the x and sine values using Matplotlib

Label the x-axis as 'Angle [radians]' and y-axis as 'Sine value'

Add a title 'Sine Wave Visualization' to the plot

💡 Why This Matters

🌍 Real World

Scientists and engineers often need to visualize mathematical functions to understand their behavior and communicate results clearly.

💼 Career

Data analysts and scientists use SciPy and Matplotlib daily to analyze data and create visual reports that help decision-makers.

1

Create x values using NumPy linspace

Import numpy as np and create a variable called x using np.linspace to generate 100 points from 0 to 2 * pi.

SciPy

# Import numpy as np
# Create x using np.linspace from 0 to 2 * np.pi with 100 points
# Your code here

Hint

Use np.linspace(start, stop, num_points) to create evenly spaced values.

2

Calculate sine values using NumPy

Use np.sin. Create a variable called y that stores the sine of each value in x.

SciPy

import numpy as np
x = np.linspace(0, 2 * np.pi, 100)
# Calculate sine values for x and store in y
# Your code here

Hint

Use np.sin(x) to get sine values for each element in x.

3

Plot the sine wave using Matplotlib

Import matplotlib.pyplot as plt. Use plt.plot(x, y) to plot the sine wave.

SciPy

import numpy as np
x = np.linspace(0, 2 * np.pi, 100)
y = np.sin(x)
# Import matplotlib.pyplot as plt
# Plot x and y using plt.plot
# Your code here

Hint

Use plt.plot(x, y) to draw the line graph.

4

Add labels and show the plot

Use plt.xlabel to label the x-axis as 'Angle [radians]', plt.ylabel to label the y-axis as 'Sine value', and plt.title to add the title 'Sine Wave Visualization'. Finally, use plt.show() to display the plot.

SciPy

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2 * np.pi, 100)
y = np.sin(x)
plt.plot(x, y)
# Add x-axis label
# Add y-axis label
# Add plot title
# Show the plot
# Your code here

Hint

Use plt.xlabel(), plt.ylabel(), plt.title(), and plt.show() to complete the plot.

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