How To Use Numpy In Python

In this tutorial, we will learn how to efficiently perform numerical operations in Python using the NumPy library. NumPy, short for Numerical Python, is a core data manipulation and scientific computing package widely used for high-performance mathematical operations. Its main feature is the n-dimensional array, or the ndarray object, allowing for efficient array-based computation.

Let’s get started by learning how to install and import NumPy, basic operations, and useful functions.

Step 1: Install and Import NumPy

If you haven’t already installed NumPy, you can do so using pip with the following command:

1
pip install numpy

Once NumPy is installed, you can import it into your Python script using the following command:

1
import numpy as np

By convention, NumPy is typically imported under the alias np. This is not required, but it’s widely used within the Python community.

Step 2: Creating NumPy Arrays

There are several ways to create a NumPy array. Let’s explore some of the most common methods.

From Python Lists

To create a NumPy array from a Python list, you can use the np.array() function:

1
2
3
4
import numpy as np
 
python_list = [1, 2, 3, 4, 5]
numpy_array = np.array(python_list)

Using Built-in Functions

NumPy offers several built-in functions to help create specific types of arrays:

  1. np.zeros(): creates an array filled with zeros
  2. np.ones(): creates an array filled with ones
  3. np.arange(): creates an array of evenly spaced values within a given interval
  4. np.linspace(): creates an array of evenly spaced values over a specified range

Here are some examples:

1
2
3
4
5
6
import numpy as np
 
zeros_array = np.zeros(5)  # Creates an array of 5 zeros
ones_array = np.ones(5)  # Creates an array of 5 ones
range_array = np.arange(0, 10, 2)  # Creates an array of values from 0 to 10 (exclusive) with a step of 2
linspace_array = np.linspace(0, 10, 5)  # Creates an array of 5 equally spaced values between 0 and 10 (inclusive)

Step 3: Basic Array Operations

With NumPy arrays, you can perform element-wise mathematical operations with ease:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import numpy as np
 
array1 = np.array([2, 4, 6])
array2 = np.array([1, 3, 5])
 
# Addition
sum_array = array1 + array2
print("Addition:", sum_array)
 
# Subtraction
difference_array = array1 - array2
print("Subtraction:", difference_array)
 
# Multiplication
product_array = array1 * array2
print("Multiplication:", product_array)
 
# Division
quotient_array = array1 / array2
print("Division:", quotient_array)
 

Step 4: Useful NumPy Functions

NumPy provides numerous mathematical functions to manipulate arrays. Here are some examples:

Reshaping Arrays

You can use np.reshape() to change the shape of an array without altering the data:

1
2
3
4
import numpy as np
 
array = np.arange(1, 10)  # Creates an array of values from 1 to 9
reshaped_array = np.reshape(array, (3, 3))  # Reshape the array to a 3 x 3 matrix

Transposing Arrays

Using np.transpose() or the .T attribute, you can transpose an array:

1
2
3
4
5
import numpy as np
 
array = np.arange(1, 10).reshape(3, 3)  # Create a 3x3 array with values from 1 to 9
transposed_array = np.transpose(array)  # Transpose the array
transposed_array2 = array.T  # Another way to transpose the array

Statistical Functions

NumPy offers various statistical functions, such as np.sum(), np.mean(), np.min(), and np.max():

1
2
3
4
5
6
7
8
import numpy as np
 
array = np.array([1, 2, 3, 4, 5])
 
total = np.sum(array)
mean = np.mean(array)
minimum = np.min(array)
maximum = np.max(array)

That’s it! With this tutorial, you should now have a basic understanding of how to use NumPy in Python for efficient numerical operations. NumPy is a powerful library with many more features. As you practice, you’re likely to discover even more efficient ways to use NumPy in your data manipulation and scientific computing tasks.