Scribd
Upload
21 views4 pages

NumPy Array Search Techniques

The document explains how to search arrays in NumPy using the where() method to find indexes of specific values and the searchsorted() method for determining where values should be inserted in sorted arrays. It provides examples for finding indexes of specific values, even and odd numbers, and inserting multiple values while maintaining order. Additionally, it describes how to search from the left or right side of the array.

Uploaded by

dhillonharjotsingh09
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
21 views4 pages

NumPy Array Search Techniques

The document explains how to search arrays in NumPy using the where() method to find indexes of specific values and the searchsorted() method for determining where values should be inserted in sorted arrays. It provides examples for finding indexes of specific values, even and odd numbers, and inserting multiple values while maintaining order. Additionally, it describes how to search from the left or right side of the array.

Uploaded by

dhillonharjotsingh09
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd

NumPy Searching Arrays

[Link]/python/numpy/numpy_array_search.asp

❮ Previous Next ❯

Searching Arrays
You can search an array for a certain value, and return the indexes that get a match.

To search an array, use the where() method.

Example

Find the indexes where the value is 4:

import numpy as np

arr = [Link]([1, 2, 3, 4, 5, 4, 4])

x = [Link](arr == 4)

print(x)
Try it Yourself »
The example above will return a tuple: (array([3, 5, 6],)

Which means that the value 4 is present at index 3, 5, and 6.

Example

Find the indexes where the values are even:

import numpy as np

arr = [Link]([1, 2, 3, 4, 5, 6, 7, 8])

1/4
x = [Link](arr%2 == 0)

print(x)
Try it Yourself »

Example
Find the indexes where the values are odd:

import numpy as np

arr = [Link]([1, 2, 3, 4, 5, 6, 7, 8])

x = [Link](arr%2 == 1)

print(x)
Try it Yourself »

Search Sorted
There is a method called searchsorted() which performs a binary search in the array,
and returns the index where the specified value would be inserted to maintain the search
order.

The searchsorted() method is assumed to be used on sorted arrays.

Example

Find the indexes where the value 7 should be inserted:

import numpy as np

arr = [Link]([6, 7, 8, 9])

x = [Link](arr, 7)

print(x)
Try it Yourself »
Example explained: The number 7 should be inserted on index 1 to remain the sort order.

The method starts the search from the left and returns the first index where the number 7
is no longer larger than the next value.

Search From the Right Side


By default the left most index is returned, but we can give side='right' to return the
right most index instead.

2/4
Example

Find the indexes where the value 7 should be inserted, starting from the right:

import numpy as np

arr = [Link]([6, 7, 8, 9])

x = [Link](arr, 7, side='right')

print(x)
Try it Yourself »
Example explained: The number 7 should be inserted on index 2 to remain the sort order.

The method starts the search from the right and returns the first index where the number
7 is no longer less than the next value.

Multiple Values
To search for more than one value, use an array with the specified values.

Example

Find the indexes where the values 2, 4, and 6 should be inserted:

import numpy as np

arr = [Link]([1, 3, 5, 7])

x = [Link](arr, [2, 4, 6])

print(x)
Try it Yourself »
The return value is an array: [1 2 3] containing the three indexes where 2, 4, 6 would be
inserted in the original array to maintain the order.

❮ Previous Next ❯

3/4
★ +1

4/4

You might also like