Array Operations in NumPy

Duration: 5 min

This module delves into the fundamental array operations in NumPy, a powerful library for numerical computing in Python. Understanding these operations is crucial for data manipulation, mathematical computations, and efficient data processing, which are essential skills in data science.

Creating and Initializing Arrays

NumPy arrays are the core data structure in NumPy. They are homogeneous, multidimensional arrays that allow for efficient storage and manipulation of data. Arrays can be created from lists, tuples, or by using various NumPy functions like numpy.array, numpy.zeros, numpy.ones, and numpy.arange.

import numpy as np

# Creating an array from a list
arr1 = np.array([1, 2, 3, 4])

# Creating an array of zeros
arr2 = np.zeros((2, 3))

# Creating an array of ones
arr3 = np.ones((3, 2))

# Creating an array with a range of values
arr4 = np.arange(0, 10, 2)

print(arr1)
print(arr2)
print(arr3)
print(arr4)

Try it in Google Colab:

[1 2 3 4]
[[0. 0. 0.]
 [0. 0. 0.]]
[[1. 1.]
 [1. 1.]
 [1. 1.]]
[0 2 4 6 8]

Basic Array Operations

NumPy provides a wide range of operations that can be performed on arrays, including arithmetic operations, aggregation functions, and element-wise operations. These operations are highly optimized and allow for efficient computation on large datasets.

import numpy as np

# Creating two arrays
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# Element-wise addition
addition = arr1 + arr2

# Element-wise multiplication
multiplication = arr1 * arr2

# Aggregation: sum of elements
sum_arr1 = np.sum(arr1)

print(addition)
print(multiplication)
print(sum_arr1)

💡 Tip: When performing operations on arrays, ensure that the arrays have compatible shapes to avoid broadcasting errors.

Key Concepts

Check Your Understanding