Why sorting matters in NumPy - Challenge Your Understanding

Challenge - 5 Problems

❓ Predict Output

intermediate

Output of sorting a NumPy array

What is the output of this code that sorts a NumPy array?

NumPy

import numpy as np
arr = np.array([3, 1, 4, 1, 5])
sorted_arr = np.sort(arr)
print(sorted_arr)

A [5 4 3 1 1]

B [3 1 4 1 5]

C [1 1 3 4 5]

D [1 3 4 5]

💡 Hint

Think about what sorting does to the order of elements.

np.sort returns a new array with elements in ascending order. The original array stays the same.

❓ data_output

intermediate

Number of elements after sorting with unique

What is the number of unique elements after sorting this NumPy array?

NumPy

import numpy as np
arr = np.array([2, 3, 2, 5, 3, 7])
unique_sorted = np.unique(arr)
print(len(unique_sorted))

A 4

B 5

C 6

D 3

💡 Hint

Unique elements are counted after sorting and removing duplicates.

np.unique returns sorted unique elements. The array has 4 unique values: 2, 3, 5, 7.

🔧 Debug

advanced

Identify the error in sorting a 2D array

What error does this code raise when trying to sort a 2D NumPy array incorrectly?

NumPy

import numpy as np
arr = np.array([[3, 2], [1, 4]])
sorted_arr = np.sort(arr, axis=2)
print(sorted_arr)

A ValueError: could not broadcast input array

B IndexError: axis 2 is out of bounds for array of dimension 2

C TypeError: sort() missing required argument 'axis'

D No error, prints sorted array

💡 Hint

Check the dimensions of the array and the axis parameter.

The array has 2 dimensions (axes 0 and 1). Axis 2 does not exist, causing IndexError.

🚀 Application

advanced

Sorting to find median in NumPy

Which code correctly finds the median of a NumPy array by sorting?

A median = np.mean(np.sort(arr))

B median = np.median(arr)

C median = np.sort(arr)[len(arr)//2]

D

sorted_arr = np.sort(arr)
median = sorted_arr[len(arr)//2]

💡 Hint

Median is the middle value after sorting.

Option D sorts the array and picks the middle element correctly. Option D uses np.median but does not show sorting. Option D is similar but misses storing sorted array. Option D calculates mean, not median.

🧠 Conceptual

expert

Why sorting is important in data science

Why is sorting data important before applying many data science algorithms?

A Sorting organizes data to enable efficient searching and grouping.

B Sorting changes data values to improve model accuracy.

C Sorting removes duplicates automatically from data.

D Sorting encrypts data to protect privacy.

💡 Hint

Think about how sorted data helps algorithms work faster or better.

Sorting arranges data in order, which helps algorithms like binary search, grouping, and visualization work efficiently. It does not change values, remove duplicates, or encrypt data.

Practice

1. Why is sorting data important in data analysis using numpy?

easy

A. It helps organize data to find trends and top values easily.

B. It deletes duplicate values automatically.

C. It changes the data type of the array elements.

D. It increases the size of the dataset.

Solution

1. Step 1: Understand sorting purpose

   Sorting arranges data in order, making it easier to analyze and find patterns.

2. Step 2: Identify correct effect of sorting

   Sorting does not delete duplicates or change data types; it only orders data.

3. Final Answer:

   It helps organize data to find trends and top values easily. -> Option A

4. Quick Check:

   Sorting = Organizing data for analysis [OK]

Hint: Sorting arranges data to spot patterns fast [OK]

Common Mistakes:

• Thinking sorting removes duplicates
• Believing sorting changes data types
• Assuming sorting increases data size

2. Which of the following is the correct syntax to return a sorted copy of a 1D numpy array named arr?

easy

A. numpy.sort(arr)

B. arr.sort(numpy)

C. sort.numpy(arr)

D. arr.sort()

Solution

1. Step 1: Recall numpy sorting syntax

   The function numpy.sort() is used to sort arrays and takes the array as argument.

2. Step 2: Evaluate the options

   arr.sort() sorts in place and returns None. arr.sort(numpy) and sort.numpy(arr) are invalid syntax. numpy.sort(arr) returns a sorted copy.

3. Final Answer:

   numpy.sort(arr) -> Option A

4. Quick Check:

   Correct syntax = numpy.sort(arr) [OK]

Hint: Use numpy.sort(array) to sort arrays [OK]

Common Mistakes:

• Using arr.sort() which sorts in place
• Using arr.sort(numpy) which is invalid
• Writing sort.numpy(arr) which is invalid

3. What is the output of the following code?

import numpy as np
arr = np.array([3, 1, 4, 1, 5])
sorted_arr = np.sort(arr)
print(sorted_arr)

medium

A. [5 4 3 1 1]

B. [3 1 4 1 5]

C. [1 1 3 4 5]

D. [1 3 4 5]

Solution

1. Step 1: Understand np.sort() behavior

   np.sort() returns a sorted copy of the array in ascending order.

2. Step 2: Sort the array values

   Original array is [3, 1, 4, 1, 5]. Sorted ascending is [1, 1, 3, 4, 5].

3. Final Answer:

   [1 1 3 4 5] -> Option C

4. Quick Check:

   np.sort([3,1,4,1,5]) = [1 1 3 4 5] [OK]

Hint: np.sort() returns ascending sorted array [OK]

Common Mistakes:

• Confusing ascending with descending order
• Expecting original array to change
• Missing duplicate values in output

4. The code below is intended to sort a 2D numpy array by rows, but it raises an error. What is the problem?

import numpy as np
arr = np.array([[3, 2], [1, 4]])
sorted_arr = np.sort(arr, axis=2)
print(sorted_arr)

medium

A. np.sort() cannot sort 2D arrays.

B. Axis 2 does not exist for a 2D array.

C. The array must be flattened before sorting.

D. The print statement is incorrect.

Solution

1. Step 1: Check array dimensions

   The array shape is (2, 2), so it has axes 0 and 1 only.

2. Step 2: Understand axis parameter in np.sort()

   Axis=2 is invalid because the array has no third axis, causing an error.

3. Final Answer:

   Axis 2 does not exist for a 2D array. -> Option B

4. Quick Check:

   Axis must be 0 or 1 for 2D arrays [OK]

Hint: Check array shape before choosing axis [OK]

Common Mistakes:

• Using axis value outside array dimensions
• Thinking np.sort can't handle 2D arrays
• Assuming print statement causes error

5. You have a 2D numpy array representing exam scores of students:

import numpy as np
scores = np.array([[88, 92, 79], [95, 85, 91], [70, 78, 88]])

How would sorting each student's scores help in quickly finding their median score?

hard

A. Sorting changes scores to percentages.

B. Sorting removes the lowest and highest scores automatically.

C. Sorting combines all scores into one list.

D. Sorting arranges scores so the middle value is easy to pick as median.

Solution

1. Step 1: Understand median calculation

   The median is the middle value in sorted data.

2. Step 2: Role of sorting in median

   Sorting each student's scores orders them, making it easy to pick the middle score as median.

3. Final Answer:

   Sorting arranges scores so the middle value is easy to pick as median. -> Option D

4. Quick Check:

   Median needs sorted data [OK]

Hint: Sort to find median easily [OK]

Common Mistakes:

• Thinking sorting removes scores
• Confusing sorting with scaling scores
• Assuming sorting merges all data