np.sqrt() for square roots in NumPy - Step-by-Step Execution

Concept Flow - np.sqrt() for square roots

Start with array or number

↓

Call np.sqrt() function

↓

Calculate square root element-wise

↓

Return new array with roots

↓

End

The np.sqrt() function takes a number or array and returns the square root of each element.

Execution Sample

NumPy

import numpy as np
arr = np.array([4, 9, 16])
roots = np.sqrt(arr)
print(roots)

Calculate square roots of each element in the array [4, 9, 16].

Execution Table

Step	Input Value	Action	Output Value
1	4	Calculate sqrt(4)	2.0
2	9	Calculate sqrt(9)	3.0
3	16	Calculate sqrt(16)	4.0
4	[4,9,16]	Apply sqrt element-wise	[2.0, 3.0, 4.0]
5	Print roots	Output array	[2.0 3.0 4.0]

💡 All elements processed, square roots calculated and printed.

Variable Tracker

Variable	Start	After Step 1	After Step 2	After Step 3	Final
arr	undefined	[4, 9, 16]	[4, 9, 16]	[4, 9, 16]	[4, 9, 16]
roots	undefined	undefined	undefined	[2.0, 3.0, 4.0]	[2.0, 3.0, 4.0]

Key Moments - 2 Insights

Why does np.sqrt() return a float even if the input is an integer?

Square roots often are not whole numbers, so np.sqrt() returns floats to keep decimal precision, as shown in execution_table steps 1-3.

What happens if np.sqrt() is given a negative number?

np.sqrt() returns nan or a warning for negative inputs because square root of negative numbers is not defined in real numbers. This is not shown here but important to know.

Visual Quiz - 3 Questions

Look at the execution table, what is the output of sqrt(9) at step 2?

A 3.0

B 9

C 4.5

D 2.0

💡 Hint

Check the 'Output Value' column at step 2 in the execution_table.

At which step is the entire array of square roots created?

A Step 3

B Step 4

C Step 1

D Step 5

💡 Hint

Look for the step where sqrt is applied element-wise to the whole array.

If the input array had a negative number, what would likely happen to the output?

A It would return a complex number array

B It would ignore the negative number

C It would return nan or warning for that element

D It would return zero for that element

💡 Hint

Recall the key moment about negative inputs and np.sqrt() behavior.

Concept Snapshot

np.sqrt(x) computes the square root of x.
Input can be a number or numpy array.
Returns float results element-wise for arrays.
Negative inputs cause nan or warnings.
Useful for math and data analysis.

Full Transcript

This visual shows how np.sqrt() works step-by-step. We start with an array of numbers. Each number is processed to find its square root, which is usually a float. The results are collected into a new array. Finally, the array of roots is printed. This helps understand how numpy handles element-wise math functions.