Bird
Raised Fist0
NumPydata~5 mins

Random choice from array in NumPy

Choose your learning style10 modes available

Start learning this pattern below

Jump into concepts and practice - no test required

or
Recommended
Test this pattern10 questions across easy, medium, and hard to know if this pattern is strong
Introduction

We use random choice to pick one or more items from a list or array without any specific order. This helps when we want to simulate random events or select samples.

Choosing a random winner from a list of participants in a contest.
Selecting random samples from a dataset for testing or training.
Simulating dice rolls or card draws in games.
Randomly picking a product to recommend to a user.
Testing algorithms with random inputs.
Syntax
NumPy
import numpy as np

# To pick one random element
random_element = np.random.choice(array)

# To pick multiple random elements with replacement
random_elements = np.random.choice(array, size=number_of_elements, replace=True)

# To pick multiple random elements without replacement
random_elements_no_repeat = np.random.choice(array, size=number_of_elements, replace=False)

The array can be a list or a numpy array.

replace=True means the same element can be picked more than once.

Examples
Randomly picks one element from the list.
NumPy
import numpy as np

array = [10, 20, 30, 40, 50]

# Pick one random element
random_element = np.random.choice(array)
print(random_element)
Randomly picks three elements, elements can repeat.
NumPy
import numpy as np

array = [10, 20, 30, 40, 50]

# Pick three elements with replacement
random_elements = np.random.choice(array, size=3, replace=True)
print(random_elements)
Randomly picks three unique elements, no repeats.
NumPy
import numpy as np

array = [10, 20, 30, 40, 50]

# Pick three elements without replacement
random_elements_no_repeat = np.random.choice(array, size=3, replace=False)
print(random_elements_no_repeat)
Shows error when trying to pick from an empty array.
NumPy
import numpy as np

array = []

# What if array is empty?
try:
    random_element = np.random.choice(array)
except ValueError as error:
    print(f"Error: {error}")
Sample Program

This program shows how to pick random elements from an array in different ways using numpy.

NumPy
import numpy as np

# Create an array of numbers
numbers = [5, 10, 15, 20, 25]

print("Original array:", numbers)

# Pick one random element
random_one = np.random.choice(numbers)
print("Randomly picked one element:", random_one)

# Pick three elements with replacement
random_three_with_replace = np.random.choice(numbers, size=3, replace=True)
print("Randomly picked three elements with replacement:", random_three_with_replace)

# Pick three elements without replacement
random_three_no_replace = np.random.choice(numbers, size=3, replace=False)
print("Randomly picked three elements without replacement:", random_three_no_replace)
OutputSuccess
Important Notes

Time complexity is O(n) where n is the number of elements to pick.

Space complexity is O(n) for the output array of chosen elements.

Common mistake: Trying to pick more elements than exist without replacement causes an error.

Use replace=False when you want unique picks, replace=True when repeats are allowed.

Summary

Random choice helps pick one or more items randomly from an array.

You can pick with or without replacement depending on whether repeats are allowed.

Always check array size when picking without replacement to avoid errors.

Practice

(1/5)
1.

What does the numpy.random.choice function do?

easy
A. It picks one or more random elements from an array.
B. It sorts the array in ascending order.
C. It calculates the mean of the array elements.
D. It removes duplicate elements from the array.

Solution

  1. Step 1: Understand the function purpose

    numpy.random.choice is used to select random elements from an array.
  2. Step 2: Compare with other options

    Sorting, calculating mean, or removing duplicates are different functions and unrelated to random choice.
  3. Final Answer:

    It picks one or more random elements from an array. -> Option A
  4. Quick Check:

    Random choice = pick random elements [OK]
Hint: Random choice picks elements randomly from an array [OK]
Common Mistakes:
  • Confusing random choice with sorting
  • Thinking it calculates statistics like mean
  • Assuming it removes duplicates
2.

Which of the following is the correct syntax to pick 3 random elements from a numpy array arr with replacement?

import numpy as np
arr = np.array([10, 20, 30, 40, 50])
easy
A. np.random.choice(arr, size=3, replace=False)
B. np.random.choice(arr, 3, replace=False)
C. np.random.choice(arr, size=3, replace=True)
D. np.random.choice(arr, 3, replace=True, axis=1)

Solution

  1. Step 1: Check correct parameter names

    The function uses size to specify number of elements and replace to allow repeats.
  2. Step 2: Validate each option

    np.random.choice(arr, size=3, replace=True) uses correct parameters: size=3 and replace=True. Others have wrong parameter names or extra invalid ones.
  3. Final Answer:

    np.random.choice(arr, size=3, replace=True) -> Option C
  4. Quick Check:

    Correct syntax = size=3, replace=True [OK]
Hint: Use size= and replace= parameters correctly [OK]
Common Mistakes:
  • Using positional argument without size=
  • Setting replace=False when repeats are needed
  • Adding invalid parameters like axis
3.

What is the output of this code?

import numpy as np
np.random.seed(0)
arr = np.array([1, 2, 3, 4, 5])
result = np.random.choice(arr, size=4, replace=False)
print(result)
medium
A. [5 1 4 3]
B. [4 5 1 3]
C. [1 2 3 4]
D. [2 5 1 4]

Solution

  1. Step 1: Understand seed and choice without replacement

    Setting seed ensures reproducible random results. Choosing 4 unique elements without replacement picks 4 different values from arr.
  2. Step 2: Run code or recall output

    With seed 0, the output is [2 5 1 4].
  3. Final Answer:

    [2 5 1 4] -> Option D
  4. Quick Check:

    Seed 0 + replace=False = [2 5 1 4] [OK]
Hint: Use np.random.seed for consistent random output [OK]
Common Mistakes:
  • Ignoring seed and expecting different output
  • Confusing replace=True and replace=False results
  • Assuming output is sorted
4.

Identify the error in this code snippet:

import numpy as np
arr = np.array([1, 2, 3])
result = np.random.choice(arr, size=5, replace=False)
print(result)
medium
A. Size is larger than array length without replacement, causing an error.
B. Array is not defined properly.
C. Missing import statement for numpy.
D. replace parameter should be True to avoid error.

Solution

  1. Step 1: Check size vs array length with replace=False

    Choosing 5 elements without replacement from an array of length 3 is invalid and raises an error.
  2. Step 2: Confirm other parts are correct

    Array is defined, numpy is imported, and replace parameter is valid. The main issue is size too large without replacement.
  3. Final Answer:

    Size is larger than array length without replacement, causing an error. -> Option A
  4. Quick Check:

    Size > array length + replace=False = error [OK]
Hint: Size must not exceed array length if replace=False [OK]
Common Mistakes:
  • Ignoring size vs array length mismatch
  • Assuming replace=True by default
  • Not importing numpy before use
5.

You have a numpy array data = np.array([10, 20, 30, 40, 50]). You want to randomly select 3 unique elements but ensure the number 20 is always included in the result. Which approach is correct?

hard
A. Pick 3 elements with replacement and filter for 20 after.
B. Remove 20 from array, pick 2 without replacement, then add 20 back.
C. Use np.random.choice(data, size=3, replace=False) directly.
D. Pick 3 elements with replacement and add 20 manually.

Solution

  1. Step 1: Understand the requirement

    We want 3 unique elements including 20 always.
  2. Step 2: Choose method to guarantee 20

    Remove 20, pick 2 unique elements from remaining, then add 20 to result ensures 20 is included and no duplicates.
  3. Step 3: Evaluate other options

    Direct random choice may exclude 20. Picking with replacement can cause duplicates or exclude 20. Adding 20 manually after picking with replacement may cause duplicates.
  4. Final Answer:

    Remove 20 from array, pick 2 without replacement, then add 20 back. -> Option B
  5. Quick Check:

    Guarantee element by picking rest then adding it [OK]
Hint: Pick others without 20, then add 20 to ensure inclusion [OK]
Common Mistakes:
  • Assuming random choice always includes 20
  • Using replacement causing duplicates
  • Adding 20 after picking with replacement causing repeats