Why linear algebra matters in NumPy - See It in Action

Why linear algebra matters

📖 Scenario: Imagine you work at a small store that sells three types of fruits: apples, bananas, and oranges. You want to understand how many fruits you sold in total each day and how much money you made. Linear algebra helps you organize and calculate this information quickly using arrays.

🎯 Goal: You will create arrays to represent daily fruit sales and prices, then use linear algebra (matrix multiplication) to find total sales and total revenue.

📋 What You'll Learn

Use numpy arrays to store data

Use matrix multiplication to calculate totals

Print the total fruits sold each day and total revenue

💡 Why This Matters

🌍 Real World

Stores and businesses use linear algebra to quickly calculate sales totals and revenue from multiple products over time.

💼 Career

Data analysts and scientists use these techniques to summarize and analyze sales data efficiently.

1

Create the sales data array

Create a numpy array called sales with these exact values representing fruits sold over 3 days: Day 1: 10 apples, 5 bananas, 8 oranges; Day 2: 7 apples, 3 bananas, 6 oranges; Day 3: 12 apples, 8 bananas, 5 oranges.

NumPy

import numpy as np
# Create the sales array with the exact values
# Your code here

Hint

Use np.array and enter the data as a list of lists, each inner list for one day.

2

Create the price array

Create a numpy array called prices with these exact values representing the price per fruit: 2 for apples, 1 for bananas, and 1.5 for oranges.

NumPy

import numpy as np
sales = np.array([[10, 5, 8], [7, 3, 6], [12, 8, 5]])
# Create the prices array with the exact values
# Your code here

Hint

Use np.array with a single list for prices.

3

Calculate total fruits sold each day

Create a numpy array called total_fruits by summing the sales array across the fruit columns (axis 1).

NumPy

import numpy as np
sales = np.array([[10, 5, 8], [7, 3, 6], [12, 8, 5]])
prices = np.array([2, 1, 1.5])
# Calculate total fruits sold each day
# Your code here

Hint

Use the sum method with axis=1 to add across columns (fruits).

4

Calculate total revenue and print results

Create a numpy array called total_revenue by multiplying sales with prices using matrix multiplication. Then print total_fruits and total_revenue.

NumPy

import numpy as np
sales = np.array([[10, 5, 8], [7, 3, 6], [12, 8, 5]])
prices = np.array([2, 1, 1.5])
total_fruits = sales.sum(axis=1)
# Calculate total revenue using matrix multiplication
# Print total_fruits and total_revenue
# Your code here

Hint

Use the @ operator for matrix multiplication. Then use print to show results.

Practice

1.

Why is linear algebra important in data science when using numpy?

easy

A. It replaces the need for any programming language.

B. It is used only for creating visualizations.

C. It helps handle and transform large sets of numbers efficiently.

D. It is only useful for text data processing.

Solution

1. Step 1: Understand the role of linear algebra

   Linear algebra allows us to work with vectors and matrices, which represent many numbers at once.

2. Step 2: Connect to numpy's purpose

   NumPy uses linear algebra to efficiently perform operations on large numerical data sets.

3. Final Answer:

   It helps handle and transform large sets of numbers efficiently. -> Option C

4. Quick Check:

   Linear algebra = efficient number handling [OK]

Hint: Linear algebra = fast math with many numbers [OK]

Common Mistakes:

• Thinking linear algebra is only for visuals
• Believing it replaces programming
• Assuming it only works with text

2.

Which of the following is the correct way to create a 2x2 matrix using numpy?

import numpy as np
matrix = ?

easy

A. np.array([[1, 2], 3, 4])

B. np.array([[1, 2], [3, 4]])

C. np.array(1, 2, 3, 4)

D. np.matrix([1, 2, 3, 4])

Solution

1. Step 1: Recall numpy array syntax for matrices

   A 2x2 matrix requires a list of lists, each inner list is a row.

2. Step 2: Check each option's structure

   np.array([[1, 2], [3, 4]]) uses nested lists correctly; others do not form a proper 2x2 matrix.

3. Final Answer:

   np.array([[1, 2], [3, 4]]) -> Option B

4. Quick Check:

   Nested lists = matrix shape [OK]

Hint: Use nested lists for matrix shape [OK]

Common Mistakes:

• Using flat lists instead of nested
• Missing brackets around rows
• Confusing np.matrix with np.array

3.

What is the output of this code?

import numpy as np
A = np.array([[1, 2], [3, 4]])
B = np.array([[2, 0], [1, 2]])
result = np.dot(A, B)
print(result)

medium

A. [[1 2]
[3 4]]

B. [[2 0]
[1 2]]

C. [[3 4]
[4 6]]

D. [[4 4]
[10 8]]

Solution

1. Step 1: Understand matrix multiplication with np.dot

   np.dot multiplies matrices by summing products of rows and columns.

2. Step 2: Calculate each element of result

   First row, first column: 1*2 + 2*1 = 4; first row, second column: 1*0 + 2*2 = 4; second row, first column: 3*2 + 4*1 = 10; second row, second column: 3*0 + 4*2 = 8.

3. Final Answer:

   [[4 4] [10 8]] -> Option D

4. Quick Check:

   Matrix multiplication = [[4 4], [10 8]] [OK]

Hint: Multiply rows by columns, sum products [OK]

Common Mistakes:

• Adding matrices instead of multiplying
• Confusing element-wise with dot product
• Mixing up row and column indices

4.

Find the error in this code snippet that tries to multiply two matrices:

import numpy as np
A = np.array([[1, 2, 3], [4, 5, 6]])
B = np.array([[7, 8], [9, 10]])
result = np.dot(A, B)
print(result)

medium

A. Matrix dimensions do not align for multiplication.

B. np.dot is not the correct function for multiplication.

C. Arrays A and B must be the same shape.

D. The print statement syntax is incorrect.

Solution

1. Step 1: Check shapes of matrices A and B

   A is 2x3, B is 2x2; for multiplication, columns of A must equal rows of B.

2. Step 2: Identify mismatch

   Since A has 3 columns and B has 2 rows, multiplication is not possible.

3. Final Answer:

   Matrix dimensions do not align for multiplication. -> Option A

4. Quick Check:

   Columns A != Rows B = Error [OK]

Hint: Check matrix shapes before multiplying [OK]

Common Mistakes:

• Ignoring shape mismatch
• Using wrong function for multiplication
• Assuming same shape needed for dot

5.

You have a dataset with 3 features and 4 samples stored as a 4x3 matrix. You want to center the data by subtracting the mean of each feature. Which numpy operation correctly achieves this?

import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
# What next?

hard

A. data - np.mean(data, axis=0)

B. data - np.mean(data, axis=1)

C. np.mean(data, axis=0) - data

D. np.mean(data, axis=1) - data

Solution

1. Step 1: Understand data shape and centering

   Data shape is 4 samples x 3 features; centering means subtracting feature means from each sample.

2. Step 2: Calculate mean along correct axis

   Axis=0 computes mean for each feature (column), which is needed to center features.

3. Step 3: Subtract feature means from data

   Subtracting np.mean(data, axis=0) from data centers each feature.

4. Final Answer:

   data - np.mean(data, axis=0) -> Option A

5. Quick Check:

   Center features by subtracting column means [OK]

Hint: Subtract mean along columns (axis=0) to center features [OK]

Common Mistakes:

• Using axis=1 subtracts row means, not features
• Subtracting data from mean reverses centering
• Confusing samples and features axes