Change language

# numpy.where () in Python

| |

The numpy.where (condition [, x, y]) function returns the indices of the elements in the input array where the given condition is met.

Parameters:
condition: When True, yield x, otherwise yield y.
x, y: Values ​​from which to choose. x, y and condition need to be broadcastable to some shape.

Returns:
out: [ndarray or tuple of ndarrays] If both x and y are specified, the output array contains elements of x where condition is True, and elements from y elsewhere.

If only condition is given, return the tuple condition.nonzero (), the indices where condition is True.

Code # 1:

` `

` # Python program explaining # where () function   import numpy as np   np.where ([[ True , False ], [  True , True ]], [[[ 1 , 2 ], [ 3 , 4 ]], [[ 5 , 6 ], [ 7 , 8 ] ]) `

Output:

` array ([[1, 6], [3, 4]]) `

Code # 2:

 ` # Python program explaining ` ` # where () function ` ` `  ` import ` ` numpy as np `   ` # a is an array of integers. ` ` a ` ` = ` ` np.array ([[` ` 1 ` `, ` ` 2 ` `, ` ` 3 ` `], [` ` 4 ` `, ` ` 5 ` `, ` ` 6 ` `]]) `   ` print ` ` (a) `   ` print ` ` (` ` ’Indices of elements "4’ ` `) `   ` b ` ` = ` ` np.where (a "` ` 4 ` `) ` ` print ` ` (b) `   ` print ` ` (` ` "Elements which are "4" ` `) ` ` print ` ` (a [b]) `

Output:

` [[1 2 3] [4 5 6]] Indices of elements "4 (array ([0 , 0, 0], dtype = int64), array ([0, 1, 2], dtype = int64)) Elements which are "4 array ([1, 2, 3]) `

## Shop

Learn programming in R: courses

\$

Best Python online courses for 2022

\$

Best laptop for Fortnite

\$

Best laptop for Excel

\$

Best laptop for Solidworks

\$

Best laptop for Roblox

\$

Best computer for crypto mining

\$

Best laptop for Sims 4

\$

Latest questions

NUMPYNUMPY

Common xlabel/ylabel for matplotlib subplots

NUMPYNUMPY

How to specify multiple return types using type-hints

NUMPYNUMPY

Why do I get "Pickle - EOFError: Ran out of input" reading an empty file?

NUMPYNUMPY

Flake8: Ignore specific warning for entire file

NUMPYNUMPY

glob exclude pattern

NUMPYNUMPY

How to avoid HTTP error 429 (Too Many Requests) python

NUMPYNUMPY

Python CSV error: line contains NULL byte

NUMPYNUMPY

csv.Error: iterator should return strings, not bytes

## Wiki

Python | How to copy data from one Excel sheet to another

Common xlabel/ylabel for matplotlib subplots

Check if one list is a subset of another in Python

How to specify multiple return types using type-hints

Printing words vertically in Python

Python Extract words from a given string

Cyclic redundancy check in Python

Finding mean, median, mode in Python without libraries