The foundation of NumPy is the ndarray, a fast and memory-efficient alternative to Python lists for numerical data. NumPy provides a rich set of functions for generating these arrays in various ways, catering to different needs in data analysis and machine learning.
The most straightforward way to create a NumPy array is by converting existing Python sequence-like objects, such as lists or tuples, using the np.array() function.
import numpy as np
# Creating a 1-dimensional array from a Python list
list_data = [1, 2, 3, 4, 5]
arr1d = np.array(list_data)
print(arr1d)
# Output: [1 2 3 4 5]
print(arr1d.dtype) # Check the data type
# Output: int64 (or int32 depending on your system)
# Creating a 2-dimensional array from a list of lists
nested_list = [[1, 2, 3], [4, 5, 6]]
arr2d = np.array(nested_list)
print(arr2d)
# Output:
# [[1 2 3]
# [4 5 6]]
print(arr2d.shape) # Check the dimensions (rows, columns)
# Output: (2, 3)
NumPy attempts to infer the most appropriate data type (dtype) for the array upon creation. However, you can explicitly specify the data type using the dtype argument. This is important for controlling memory usage and numerical precision.
# Specifying float data type
arr_float = np.array([1, 2, 3], dtype=np.float64)
print(arr_float)
# Output: [1. 2. 3.]
print(arr_float.dtype)
# Output: float64
# Specifying boolean data type
arr_bool = np.array([0, 1, 2, 0, 3], dtype=bool)
print(arr_bool)
# Output: [False True True False True]
print(arr_bool.dtype)
# Output: bool
Remember that NumPy arrays are homogeneous; all elements must be of the same data type. If you provide data with mixed types (e.g., integers and floats), NumPy will upcast them to the most general type that can accommodate all elements (usually float or object).
Often, you need to create an array of a specific size and shape without knowing the final values yet, perhaps as a placeholder to be filled later. NumPy offers several functions for this:
np.zeros(): Creates an array filled entirely with zeros.np.ones(): Creates an array filled entirely with ones.np.full(): Creates an array filled with a specified constant value.np.empty(): Creates an array whose initial content is random and depends on the state of the memory. It's slightly faster than zeros or ones as it avoids filling the array, but you must explicitly assign values to every element before using it.All these functions take a shape argument, which is typically a tuple specifying the dimensions of the array (e.g., (rows, columns) for a 2D array), and an optional dtype argument.
# Create a 3x4 array of zeros (defaults to float64)
zeros_arr = np.zeros((3, 4))
print(zeros_arr)
# Output:
# [[0. 0. 0. 0.]
# [0. 0. 0. 0.]
# [0. 0. 0. 0.]]
# Create a 1D array of 5 ones with integer type
ones_arr = np.ones(5, dtype=np.int32)
print(ones_arr)
# Output: [1 1 1 1 1]
# Create a 2x2 array filled with the value 99
full_arr = np.full((2, 2), 99)
print(full_arr)
# Output:
# [[99 99]
# [99 99]]
# Create an uninitialized 2x3 array
empty_arr = np.empty((2, 3))
print(empty_arr) # Values will be arbitrary
# Output (example, will vary):
# [[6.95190771e-310 6.95190771e-310 6.95190771e-310]
# [6.95190771e-310 0.00000000e+000 0.00000000e+000]]
These functions are frequently used in machine learning, for example, to initialize weight matrices before training a model or to create result arrays.
NumPy provides functions analogous to Python's built-in range function but designed to produce NumPy arrays directly:
np.arange(): Returns evenly spaced values within a given interval. It takes start, stop, and step arguments, similar to range. Note that the stop value is exclusive.np.linspace(): Returns evenly spaced numbers over a specified interval. It takes start, stop, and num (the number of samples to generate) arguments. Importantly, the stop value is inclusive by default.# An array from 0 up to (but not including) 10
arr_range = np.arange(10)
print(arr_range)
# Output: [0 1 2 3 4 5 6 7 8 9]
# An array from 2 up to (but not including) 10, with a step of 2
arr_range_step = np.arange(2, 10, 2)
print(arr_range_step)
# Output: [2 4 6 8]
# 5 evenly spaced values between 0 and 1 (inclusive)
arr_linspace = np.linspace(0, 1, 5)
print(arr_linspace)
# Output: [0. 0.25 0.5 0.75 1. ]
# 10 evenly spaced values between 0 and 5 (inclusive)
arr_linspace_2 = np.linspace(0, 5, 10)
print(arr_linspace_2)
# Output: [0. 0.55555556 1.11111111 1.66666667 2.22222222 2.77777778
# 3.33333333 3.88888889 4.44444444 5. ]
linspace is particularly useful when you need a specific number of points distributed evenly across an interval, for example, when generating coordinates for plotting functions.
Generating arrays with random numbers is essential for various tasks in machine learning, such as initializing model parameters, creating synthetic data, or shuffling datasets. NumPy's random submodule offers a wide array of functions for this:
np.random.rand(): Creates an array of the given shape and populates it with random samples from a uniform distribution over [0, 1).np.random.randn(): Creates an array of the given shape and populates it with random samples from a standard normal distribution (mean 0 and variance 1).np.random.randint(): Returns random integers from a specified low (inclusive) to high (exclusive) boundary. You can also specify the size (shape) of the output array.np.random.seed(): Used to set the random seed, which makes the random number generation predictable. This is important for reproducibility in experiments.# Set the seed for reproducibility
np.random.seed(42)
# Create a 2x3 array with random values from a uniform distribution [0, 1)
rand_arr = np.random.rand(2, 3)
print(rand_arr)
# Output:
# [[0.37454012 0.95071431 0.73199394]
# [0.59865848 0.15601864 0.15599452]]
# Create a 3x2 array with random values from a standard normal distribution
randn_arr = np.random.randn(3, 2)
print(randn_arr)
# Output:
# [[ 0.05808361 -0.75634998]
# [-0.34791215 0.1579198 ]
# [ 0.45615031 0.99712472]]
# Generate 5 random integers between 1 (inclusive) and 10 (exclusive)
randint_arr = np.random.randint(1, 10, size=5)
print(randint_arr)
# Output: [8 4 6 8 8]
# Generate a 2x4 array of random integers between 0 (inclusive) and 5 (exclusive)
randint_arr_2d = np.random.randint(0, 5, size=(2, 4))
print(randint_arr_2d)
# Output:
# [[3 0 2 3]
# [1 1 4 2]]
NumPy also includes functions for creating specific types of arrays:
np.eye(): Creates a 2D identity matrix (1s on the diagonal, 0s elsewhere).np.diag(): Can either extract the diagonal of an existing 2D array or create a 2D array with specified values on the diagonal.# Create a 3x3 identity matrix
identity_matrix = np.eye(3)
print(identity_matrix)
# Output:
# [[1. 0. 0.]
# [0. 1. 0.]
# [0. 0. 1.]]
# Create a 2D array with [1, 2, 3] on the diagonal
diag_matrix = np.diag([1, 2, 3])
print(diag_matrix)
# Output:
# [[1 0 0]
# [0 2 0]
# [0 0 3]]
Mastering these array creation techniques is the first step towards effectively using NumPy. Choosing the right method depends on whether you're converting existing data, need placeholders, require specific sequences, or need to generate random data for simulations or initializations. With these tools, you can efficiently construct the ndarray objects that form the basis for numerical operations in Python for machine learning.
Was this section helpful?
© 2026 ApX Machine LearningAI Ethics & Transparency•