Complex number type in NumPy - Mini Project: Build & Apply

Working with Complex Numbers using NumPy

📖 Scenario: Imagine you are working with signals in engineering or physics. Signals often have two parts: one part that shows the main value and another part that shows a shift or delay. These two parts can be represented as complex numbers.

🎯 Goal: You will create a NumPy array of complex numbers, set a threshold for the imaginary part, filter the numbers based on this threshold, and then print the filtered results.

📋 What You'll Learn

Create a NumPy array called complex_numbers with specific complex values

Create a variable called imag_threshold to hold a numeric threshold

Use a list comprehension to filter complex_numbers for numbers with imaginary part greater than imag_threshold

Print the filtered list of complex numbers

💡 Why This Matters

🌍 Real World

Complex numbers are used in signal processing, electrical engineering, and physics to represent waves, oscillations, and other phenomena with two components.

💼 Career

Understanding how to manipulate complex numbers with NumPy is useful for data scientists and engineers working with signals, control systems, or any domain involving complex data.

1

Create a NumPy array of complex numbers

Import NumPy as np and create a NumPy array called complex_numbers with these exact complex values: 3+4j, 1+2j, 5+0j, 0+3j, and 2+1j.

NumPy

import numpy as np
# Create the NumPy array complex_numbers with the given complex values
# Your code here

Need a hint?

Use np.array and list the complex numbers inside square brackets.

2

Set the imaginary part threshold

Create a variable called imag_threshold and set it to the number 2.

NumPy

import numpy as np
complex_numbers = np.array([3+4j, 1+2j, 5+0j, 0+3j, 2+1j])
# Create imag_threshold and set it to 2
# Your code here

Need a hint?

Just assign the number 2 to the variable imag_threshold.

3

Filter complex numbers by imaginary part

Use a list comprehension to create a list called filtered_numbers that contains only the numbers from complex_numbers whose imaginary part is greater than imag_threshold. Use num.imag to get the imaginary part of each number.

NumPy

import numpy as np
complex_numbers = np.array([3+4j, 1+2j, 5+0j, 0+3j, 2+1j])
imag_threshold = 2
# Create filtered_numbers with numbers having imaginary part > imag_threshold
# Your code here

Need a hint?

Use a list comprehension with num.imag > imag_threshold as the condition.

4

Print the filtered complex numbers

Print the variable filtered_numbers to display the filtered complex numbers.

NumPy

import numpy as np
complex_numbers = np.array([3+4j, 1+2j, 5+0j, 0+3j, 2+1j])
imag_threshold = 2
filtered_numbers = [num for num in complex_numbers if num.imag > imag_threshold]
# Print filtered_numbers
# Your code here

Need a hint?

The output should show only the complex numbers with imaginary parts greater than 2.