NumPy — Random

Random sampling

import numpy as np

# Random float [0, 1)
print(np.random.rand(3, 3))

# Random int
print(np.random.randint(0, 10, (3, 3)))

# Random normal
print(np.random.randn(3, 3))

Seed

np.random.seed(42)
arr = np.random.rand(5)
print(arr)

Random state

rng = np.random.default_rng(42)
arr = rng.random(5)
print(arr)

Distributions

# Uniform
print(np.random.uniform(0, 1, 5))

# Normal
print(np.random.normal(0, 1, 5))

# Poisson
print(np.random.poisson(5, 5))

# Exponential
print(np.random.exponential(1, 5))

Choice

arr = np.array([1, 2, 3, 4, 5])
print(np.random.choice(arr, 3, replace=False))

Shuffle

arr = np.array([1, 2, 3, 4, 5])
np.random.shuffle(arr)
print(arr)

Mini Practice

1. Generate random numbers
2. Set seed for reproducibility
3. Use different distributions
4. Practice choice and shuffle

Up Next

Continue with File I/O - Reading and writing files.