Vector Addition, Subtraction, and Scalar Multiplication with NumPy

Welcome to the first lesson of the course “Vector and Matrix Operations with NumPy.” In this lesson, we'll explore fundamental vector operations using the Python library NumPy. Python is a popular programming language known for its readability and versatility. It's widely used in various fields, including data analysis and scientific computations.

NumPy is a prominent library in Python that excels at numerical computations, especially when dealing with arrays and matrices. It's favored for its performance and ease of use, particularly for operations related to linear algebra, such as vectors and matrices.

By the end of this lesson, you'll be able to create and manipulate vectors using NumPy. Let's dive into understanding vectors in NumPy and how they can be efficiently utilized.

Before we proceed, let’s briefly remind ourselves about vectors. In mathematics, vectors are entities characterized by magnitude and direction. In NumPy, vectors are represented using arrays, which are more efficient than standard Python lists for numerical operations.

Here’s how you would represent a vector in NumPy:

import numpy as np

vector_example = np.array([1, 2, 3])

In this example, np.array([1, 2, 3]) creates a vector with values 1, 2, and 3. NumPy arrays provide powerful capabilities to perform mathematical operations efficiently, which we'll explore in this lesson.

To add two vectors in NumPy, you can use the np.add() function or simply use Python’s addition operator +. Here's an example using np.add():

vector_a = np.array([1, 2, 3])
vector_b = np.array([4, 5, 6])

addition = np.add(vector_a, vector_b)
print("Addition:", addition)

# Output:
# Addition: [5 7 9]

In this example, each element of vector_a is added to the corresponding element of vector_b, resulting in a new vector [5, 7, 9].

Similar to addition, vector subtraction can be performed with np.subtract() or the subtraction operator -. Here's how:

subtraction = np.subtract(vector_a, vector_b)
print("Subtraction:", subtraction)

# Output:
# Subtraction: [-3 -3 -3]

This example subtracts each element of vector_b from the corresponding element of vector_a, resulting in [-3, -3, -3].

Note that for both the subtraction and addition operations, they are performed only if the dimensions (shape) of the vectors are equal. This ensures that corresponding elements are properly aligned for these operations.