np.unique() for unique elements in NumPy - Step-by-Step Execution

Concept Flow - np.unique() for unique elements

Input array

↓

Sort array

↓

Find unique elements

↓

Return unique array

np.unique() takes an array, sorts it, finds unique elements, and returns them as a new array.

Execution Sample

NumPy

import numpy as np
arr = np.array([3, 1, 2, 3, 2, 4])
unique_arr = np.unique(arr)
print(unique_arr)

This code finds unique elements from the array and prints them sorted.

Execution Table

Step	Action	Input/State	Result/Output
1	Input array created	[3, 1, 2, 3, 2, 4]	[3 1 2 3 2 4]
2	Sort array internally	[3 1 2 3 2 4]	[1 2 2 3 3 4]
3	Find unique elements	[1 2 2 3 3 4]	[1 2 3 4]
4	Return unique array	-	[1 2 3 4]

💡 All unique elements extracted and returned as sorted array.

Variable Tracker

Variable	Start	After Step 1	After Step 2	After Step 3	Final
arr	-	[3 1 2 3 2 4]	[3 1 2 3 2 4]	[3 1 2 3 2 4]	[3 1 2 3 2 4]
unique_arr	-	-	-	[1 2 3 4]	[1 2 3 4]

Key Moments - 2 Insights

Why does np.unique() return a sorted array instead of preserving the original order?

np.unique() sorts the array internally to efficiently find unique elements, so the output is always sorted as shown in step 2 and 3 of the execution_table.

Does np.unique() change the original array?

No, np.unique() does not modify the original array 'arr'. It returns a new array with unique elements, as seen in variable_tracker where 'arr' stays the same.

Visual Quiz - 3 Questions

Look at the execution_table, what is the output after step 3?

A [3 1 2 3 2 4]

B [1 2 3 4]

C [1 2 2 3 3 4]

D [4 3 2 1]

💡 Hint

Check the 'Result/Output' column for step 3 in the execution_table.

At which step does np.unique() sort the array internally?

A Step 1

B Step 3

C Step 2

D Step 4

💡 Hint

Look at the 'Action' column in the execution_table for sorting.

If the input array was already sorted, how would the execution_table change?

A Step 2 would show the same sorted array as input

B Step 3 would return duplicates

C Step 4 would return the original array

D Step 1 would be skipped

💡 Hint

Refer to the sorting step and its output in the execution_table.

Concept Snapshot

np.unique(array)
- Returns sorted unique elements from array
- Does not modify original array
- Useful to find distinct values
- Output is always sorted
- Can return indices or counts with options

Full Transcript

np.unique() is a function in numpy that finds unique elements in an array. It first sorts the array internally, then extracts unique values, and returns them as a new sorted array. The original array remains unchanged. This process helps to quickly identify distinct values in data. The output is always sorted, which is important to remember when order matters.

Practice

1. What does the np.unique() function do when applied to a NumPy array?

easy

A. Returns the shape of the array

B. Returns the sum of all elements in the array

C. Returns the maximum value in the array

D. Returns all unique elements sorted from the array

Solution

1. Step 1: Understand the purpose of np.unique()

   The function np.unique() extracts all different values from the input array and sorts them.

2. Step 2: Compare with other options

   Options A, B, and D describe different functions like shape, sum, and max, which are not what np.unique() does.

3. Final Answer:

   Returns all unique elements sorted from the array -> Option D

4. Quick Check:

   np.unique() = unique sorted elements [OK]

Hint: np.unique() always returns sorted unique values [OK]

Common Mistakes:

• Thinking it returns counts by default
• Confusing with sum or max functions
• Assuming it returns unsorted unique values

2. Which of the following is the correct syntax to get unique elements from a NumPy array arr?

easy

A. np.unique(arr)

B. arr.unique()

C. unique(arr)

D. np.arr.unique()

Solution

1. Step 1: Recall the correct function call

   The correct way to call the unique function in NumPy is np.unique(arr).

2. Step 2: Identify incorrect syntax

   Options A, B, and C are invalid because either the method does not exist on the array object or the function is not called from the NumPy module correctly.

3. Final Answer:

   np.unique(arr) -> Option A

4. Quick Check:

   Use np.unique(array) syntax [OK]

Hint: Always call unique as np.unique(array) [OK]

Common Mistakes:

• Trying to call unique as a method on array
• Forgetting the np. prefix
• Using a non-existent unique() function without np

3. What is the output of the following code?

import numpy as np
arr = np.array([3, 1, 2, 3, 2, 1, 4])
result = np.unique(arr)
print(result)

medium

A. [4 3 2 1]

B. [3 1 2 4]

C. [1 2 3 4]

D. [1 1 2 2 3 3 4]

Solution

1. Step 1: Apply np.unique() to the array

   The function extracts unique values from the array: 1, 2, 3, and 4.

2. Step 2: Note the sorting behavior

   np.unique() returns these unique values sorted in ascending order: [1 2 3 4].

3. Final Answer:

   [1 2 3 4] -> Option C

4. Quick Check:

   Unique sorted values = [1 2 3 4] [OK]

Hint: np.unique() sorts unique values automatically [OK]

Common Mistakes:

• Expecting original order instead of sorted
• Including duplicates in output
• Confusing with counts output

4. The following code is intended to print unique elements of arr but raises an error. What is the error?

import numpy as np
arr = np.array([5, 6, 5, 7])
result = arr.unique()
print(result)

medium

A. 'numpy.ndarray' object has no attribute 'unique'

B. SyntaxError: invalid syntax

C. TypeError: unique() missing 1 required positional argument

D. NameError: name 'unique' is not defined

Solution

1. Step 1: Identify the method call on the array

   The code calls arr.unique(), but NumPy arrays do not have a method named unique().

2. Step 2: Understand the correct usage

   The correct function is np.unique(arr), a function in the NumPy module, not a method of the array object.

3. Final Answer:

   'numpy.ndarray' object has no attribute 'unique' -> Option A

4. Quick Check:

   Arrays have no unique() method [OK]

Hint: np.unique() is a function, not an array method [OK]

Common Mistakes:

• Calling unique() as a method on array
• Forgetting to import numpy as np
• Using wrong function name or syntax

5. Given a 2D NumPy array arr = np.array([[1, 2, 2], [3, 1, 4]]), which code correctly finds all unique elements in the entire array?

hard

A. np.unique(arr, axis=0)

B. np.unique(arr)

C. np.unique(arr, axis=1)

D. arr.unique()

Solution

1. Step 1: Understand np.unique() on 2D arrays

   Calling np.unique(arr) flattens the array and returns all unique elements sorted.

2. Step 2: Check axis parameter effects

   Using axis=0 or axis=1 returns unique rows or columns, not unique elements overall.

3. Step 3: Identify invalid method call

   arr.unique() is invalid because arrays do not have a unique method.

4. Final Answer:

   np.unique(arr) -> Option B

5. Quick Check:

   np.unique(array) finds all unique elements [OK]

Hint: Use np.unique(arr) to get all unique elements in any array [OK]

Common Mistakes:

• Using axis parameter expecting unique elements, but it returns unique rows/columns
• Calling unique() as a method on array
• Confusing unique elements with unique rows or columns