Why saving and loading matters in NumPy - See It in Action

Why Saving and Loading Matters

📖 Scenario: Imagine you are working on a data science project where you create important data arrays. You want to save your work so you can use it later without starting over.

🎯 Goal: You will create a NumPy array, save it to a file, load it back, and print it to see that your data is safe and ready to use anytime.

📋 What You'll Learn

Create a NumPy array with exact values

Save the array to a file using NumPy

Load the array from the file

Print the loaded array to confirm it matches the original

💡 Why This Matters

🌍 Real World

Data scientists often work with large datasets that take time to prepare. Saving processed data means they can pause and continue work later without losing progress.

💼 Career

Knowing how to save and load data files is essential for data science jobs to manage data efficiently and share results with others.

1

Create a NumPy array

Import NumPy as np and create a NumPy array called data with these exact values: [10, 20, 30, 40, 50].

NumPy

import numpy as np
# Create a NumPy array called data with values [10, 20, 30, 40, 50]
# Your code here

Hint

Use np.array() to create the array with the exact list of numbers.

2

Save the array to a file

Save the NumPy array data to a file named data.npy using the np.save function.

NumPy

import numpy as np
data = np.array([10, 20, 30, 40, 50])
# Save the array data to 'data.npy'
# Your code here

Hint

Use np.save(filename, array) to save the array to a file.

3

Load the array from the file

Load the array from the file data.npy into a new variable called loaded_data using the np.load function.

NumPy

import numpy as np
data = np.array([10, 20, 30, 40, 50])
np.save('data.npy', data)
# Load the array from 'data.npy' into loaded_data
# Your code here

Hint

Use np.load(filename) to load the array from the file.

4

Print the loaded array

Print the variable loaded_data to show the saved and loaded array values.

NumPy

import numpy as np
data = np.array([10, 20, 30, 40, 50])
np.save('data.npy', data)
loaded_data = np.load('data.npy')
# Print loaded_data
# Your code here

Hint

Use print(loaded_data) to display the array.

Practice

1. Why is it important to save and load NumPy arrays when working on data science projects?

easy

A. To delete your data after use

B. To make your code run slower

C. To convert arrays into strings automatically

D. To keep your data safe and reuse it without recalculating

Solution

1. Step 1: Understand the purpose of saving data

   Saving data helps keep your work safe so you don't lose it.

2. Step 2: Understand the benefit of loading data

   Loading saved data lets you reuse it without recalculating or reprocessing.

3. Final Answer:

   To keep your data safe and reuse it without recalculating -> Option D

4. Quick Check:

   Saving and loading = reuse data [OK]

Hint: Saving keeps data safe; loading reuses it fast [OK]

Common Mistakes:

• Thinking saving slows down code
• Believing saving deletes data
• Confusing saving with data conversion

2. Which of the following is the correct way to save a NumPy array named arr to a file called data.npy?

easy

A. np.save('data.npy', arr)

B. np.load('data.npy', arr)

C. np.save(arr, 'data.npy')

D. np.load(arr, 'data.npy')

Solution

1. Step 1: Identify the correct function for saving

   Use np.save to save arrays to a file.

2. Step 2: Check the argument order

   The first argument is the filename, the second is the array to save.

3. Final Answer:

   np.save('data.npy', arr) -> Option A

4. Quick Check:

   Save syntax = np.save(filename, array) [OK]

Hint: np.save(filename, array) saves data [OK]

Common Mistakes:

• Swapping filename and array arguments
• Using np.load instead of np.save to save
• Confusing save and load functions

3. What will be the output of this code?

import numpy as np
arr = np.array([1, 2, 3])
np.save('temp.npy', arr)
loaded_arr = np.load('temp.npy')
print(loaded_arr)

medium

A. Error: file not found

B. ['1' '2' '3']

C. [1 2 3]

D. [[1 2 3]]

Solution

1. Step 1: Save the array to a file

   The array [1, 2, 3] is saved to 'temp.npy' using np.save.

2. Step 2: Load the array back and print

   np.load reads the saved file and returns the original array.

3. Final Answer:

   [1 2 3] -> Option C

4. Quick Check:

   Save then load returns original array [OK]

Hint: Load after save returns original array [OK]

Common Mistakes:

• Expecting string output instead of numbers
• Thinking load reads text files
• Assuming nested array output

4. What is wrong with this code snippet?

import numpy as np
arr = np.array([4, 5, 6])
np.save('mydata.npy')
loaded = np.load('mydata.npy')
print(loaded)

medium

A. Filename should not have .npy extension

B. np.save is missing the array argument

C. Array must be saved as a list, not np.array

D. np.load cannot read .npy files

Solution

1. Step 1: Check np.save usage

   np.save requires two arguments: filename and array to save.

2. Step 2: Identify missing argument

   The code calls np.save with only filename, missing the array argument.

3. Final Answer:

   np.save is missing the array argument -> Option B

4. Quick Check:

   np.save needs filename and array [OK]

Hint: np.save needs filename and array [OK]

Common Mistakes:

• Forgetting to pass the array to np.save
• Thinking np.load can't read .npy files
• Believing .npy extension is wrong

5. You have a large NumPy array data that you want to save and share with a colleague. Which approach is best to ensure your colleague can load it exactly as you saved it, and why?

hard

A. Save with np.save and share the .npy file because it preserves array shape and data type

B. Convert array to string and save as .txt because text files are universal

C. Save with np.savez_compressed but rename file to .txt for easy sharing

D. Print array to console and ask colleague to copy-paste it

Solution

1. Step 1: Understand file formats for saving arrays

   .npy files save arrays with shape and data type intact, ideal for sharing.

2. Step 2: Evaluate options for sharing

   Text files lose shape and type info; renaming compressed files confuses loading; copy-paste is error-prone.

3. Final Answer:

   Save with np.save and share the .npy file because it preserves array shape and data type -> Option A

4. Quick Check:

   .npy files keep data exact [OK]

Hint: Use .npy files to keep array exact for sharing [OK]

Common Mistakes:

• Using text files loses array structure
• Renaming compressed files breaks loading
• Copy-pasting arrays causes errors