Skip to Content

Extract the Last N Elements of Numpy Array

The Numpy library in Python comes with a number of useful methods and techniques to work with and manipulate data in arrays. In this tutorial, we will look at how to get the last N elements of a one-dimensional Numpy Array with the help of some examples.

How to get the last n elements of a Numpy array?

last n elements of numpy array

You can use slicing to get the last n elements of a Numpy array. Slice the array from the index of the nth last element to the end of the array. Using a negative index can be helpful here. The following is a syntax –

# last n elements of numpy array
ar[-n:]

This will give us the elements in the array starting from the nth last element (the index -n represents the index of the nth element from the end of the array) to the end of the array.

Steps to get the last n elements in an array

Let’s now look at a step-by-step example of using the above syntax on a Numpy array.

Step 1 – Create a Numpy array

First, we will create a Numpy array that we’ll operate on.

import numpy as np

# create numpy array
ar = np.array([1, 5, 6, 3, 2, 4, 7])
# display the array 
print(ar)

Output:

[1 5 6 3 2 4 7]

Here, we used the numpy.array() function to create a one-dimensional Numpy array containing some numbers. There are seven elements in the array.

Step 2 – Slice the array to get the last n elements

To get the last n elements of the above array, slice the array from the index -n to the end of the array.

For example, let’s get the last 3 elements from the array that we created in step 1.

# get last 3 elements of the array
print(ar[-3:])

Output:

[2 4 7]

We get the last 3 elements of the array.

For more on slicing Numpy arrays, refer to its documentation.

Summary

In this tutorial, we looked at how to get the last n elements of a one-dimensional Numpy array using slicing. The following is a short summary of the steps mentioned in the tutorial.

  1. Create a Numpy array (skip this step if you already have an array to operate on).
  2. Slice the array from the index -n to the end of the array.

You might also be interested in –


Subscribe to our newsletter for more informative guides and tutorials.
We do not spam and you can opt out any time.


Author

  • Piyush is a data scientist passionate about using data to understand things better and make informed decisions. In the past, he's worked as a Data Scientist for ZS and holds an engineering degree from IIT Roorkee. His hobbies include watching cricket, reading, and working on side projects.