Creating and Using Python NumPy Arrays

Introduction

This lab provides a step-by-step guide to creating and using NumPy arrays. NumPy arrays are multidimensional, grid-like structures used for scientific computing and data analysis in Python. They have several advantages over Python lists, including faster computation, easier manipulation, and better memory management.

VM Tips

After the VM startup is done, click the top left corner to switch to the Notebook tab to access Jupyter Notebook for practice.

Sometimes, you may need to wait a few seconds for Jupyter Notebook to finish loading. The validation of operations cannot be automated because of limitations in Jupyter Notebook.

If you face issues during learning, feel free to ask Labby. Provide feedback after the session, and we will promptly resolve the problem for you.

Skills Graph

Installing NumPy

Before we can begin creating and using NumPy arrays, it is necessary to install the NumPy package. This can be done using the following command:

!pip install numpy

Creating a NumPy Array

To create a NumPy array, we can use the numpy.array() function. We can pass a list, tuple, or any array-like object to this function, and it will convert it into a NumPy array. The dtype parameter can be used to explicitly define the data type of the array.

import numpy as np

## create a 1-dimensional NumPy array
arr1 = np.array([1, 2, 3])

## create a 2-dimensional NumPy array
arr2 = np.array([[1, 2, 3],[4, 5, 6]])

Checking the Shape and Dimensions of a NumPy Array

We can use the shape and ndim attributes of a NumPy array to check its shape and dimensions, respectively. The shape attribute returns a tuple with the number of elements in each dimension of the array, while the ndim attribute returns the number of dimensions in the array.

import numpy as np

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

print(arr.shape) ## output: (2, 3)
print(arr.ndim) ## output: 2

Accessing Elements of a NumPy Array

We can access elements of a NumPy array using indexing and slicing, just like with Python lists. The index of a NumPy array always starts at 0.

import numpy as np

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

print(arr[0, 1]) ## output: 2
print(arr[:, 1]) ## output: array([2, 5])

Performing Operations on NumPy Arrays

NumPy arrays support many different operations, such as arithmetic operations, aggregation functions, and logical operations. For example, we can use the sum() and mean() functions to calculate the sum and mean of the elements of a NumPy array:

import numpy as np

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

print(arr.sum()) ## output: 21
print(arr.mean()) ## output: 3.5

Reshaping and Transposing NumPy Arrays

We can reshape a NumPy array using the reshape() function. This function takes a tuple of the desired shape as its argument. We can also transpose a NumPy array using the transpose() or T attribute.

import numpy as np

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

## reshape array to 3 rows and 2 columns
arr_reshaped = arr.reshape((3, 2))

## transpose array
arr_transposed = arr.transpose()
arr_T = arr.T

print(arr_reshaped) ## output: array([[1, 2],[3, 4],[5, 6]])
print(arr_transposed) ## output: array([[1, 4],[2, 5],[3, 6]])
print(arr_T) ## output: array([[1, 4],[2, 5],[3, 6]])

Summary

In this lab, we learned how to create and use NumPy arrays in Python. We covered the basic steps of creating an array, checking its shape and dimensions, accessing its elements, performing operations on it, and reshaping and transposing it. With these skills, we can efficiently work with multidimensional arrays for scientific computing and data analysis.