Lesson Overview and Plan

Hello, Data Explorer! Today's lesson will focus on "selecting data" in NumPy arrays—our goal is to master integer and Boolean indexing and slicing. So, let's jump in!

Understanding Indexing in 1D Arrays

In this section, we'll discuss "Indexing". It's akin to item numbering in lists, and Python implements it with a zero-based index system. Let's see this principle in action.

Essentially, indexing is a numeric system for items in an array—relatively straightforward!

Understanding Indexing in Multi-dimensional Arrays

Are you mindful of higher dimensions? NumPy arrays can range from 1D to N-dimensions. To access specific elements, we use the index pair (i,j) for a 2D array, (i,j,k) for a 3D array, and so forth.

In this example, we selected the first element of the second row in a 2D-array—it's pretty simple!

Understanding Boolean Indexing

Are you ready for some magic? Enter "Boolean Indexing", which functions like a 'Yes/No' filter, with 'Yes' representing True and 'No' for False.

Or we can put the condition directly into [] brackets:

Voila! Now, we can filter data based on custom conditions.

Understanding Complex Conditions in Boolean Indexing

Now that we've mastered the simple 'Yes/No' binary filter system, let's up the ante with "Complex Conditions in Boolean Indexing". This method refines our filtering process further, allowing us to set more detailed restrictions.

Imagine, for instance, that we want to create an index for even numbers greater than five. We'll merge two conditions to yield this result:

In this query, we used the ampersand (&) to signify intersection - i.e., we're selecting numbers that are both even AND larger than five. Note, that simple and operator won't work here.

Similarly, we can use the pipe operator (|) to signify union - i.e., selecting numbers that are either even OR larger than five:

Awesome, right? This additional filtering layer empowers us to be more specific and intentional about the data we select.

A Look at Slicing in NumPy Arrays

NumPy arrays could be sliced in the same manner as the regular python list. Let's make a quick recall. The syntax is start:stop:step, where start is the first index to choose (inclusive), stop is the last index to choose (exclusive), and step defines the step of the selection. For example, if the step=1, each element will be selected, and if step=2 – every other one will be skipped.

Let's take a look at simple examples:

Note that slicing is inclusive on the left and exclusive on the right.

Another example with a step parameter:

In this case, we choose every second element, by starting with 1 and using step=2.

Lesson Summary

Congratulations! We've traversed the landscape of NumPy arrays, delving into indexing, Boolean indexing, and slicing. Now, we'll dive into some hands-on exercises. After all, practice makes perfect. Let's go forth, data explorers!

Sign up
Join the 1M+ learners on CodeSignal
Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal