Introduction to NumPy

Duration: 5 min

This module introduces you to NumPy, a fundamental package for scientific computing in Python. You will learn about arrays, a powerful data structure that allows for efficient numerical computations. Understanding NumPy is crucial for data manipulation and analysis, forming the backbone of many data science tasks.

Understanding NumPy Arrays

NumPy arrays are the core data structure in NumPy. They are homogeneous, multidimensional arrays that allow for efficient storage and computation. Unlike Python lists, NumPy arrays are designed for numerical operations and support vectorized operations, which means you can perform operations on entire arrays without the need for explicit loops.

import numpy as np

# Creating a 1D array
array_1d = np.array([1, 2, 3, 4])
print('1D Array:', array_1d)

# Creating a 2D array
array_2d = np.array([[1, 2], [3, 4]])
print('2D Array:\n', array_2d)

Try it in Google Colab:

1D Array: [1 2 3 4]
2D Array:
 [[1 2]
 [3 4]]

Array Operations

NumPy arrays support a wide range of operations, including arithmetic operations, aggregations, and broadcasting. These operations are performed element-wise, making it easy to perform complex computations on large datasets.

import numpy as np

# Creating two 1D arrays
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])

# Element-wise addition
addition = array1 + array2
print('Element-wise Addition:', addition)

# Element-wise multiplication
multiplication = array1 * array2
print('Element-wise Multiplication:', multiplication)

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

Key Concepts

Check Your Understanding