logo

numpy.where() in Python

De numpy.where() functie retourneert de indices van elementen in een invoerarray waarin aan de gegeven voorwaarde is voldaan.

Syntaxis: numpy.where(voorwaarde[, x, y])
Parameters:
voorwaarde : Als dit waar is, levert u x op, anders y.
x, y: Waarden waaruit u kunt kiezen. x, y en condition moeten in een of andere vorm uitzendbaar zijn.



sorteer-java samenvoegen

Geeft terug:
uit : [ndarray of tupel van ndarrays] Als zowel x als y zijn opgegeven, bevat de uitvoerarray elementen van x waarbij de voorwaarde True is, en elementen van y elders.

Als alleen de voorwaarde is opgegeven, retourneert u de tupel condition.nonzero(), de indices waarbij de voorwaarde True is.

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>]])>

>

>

Uitgang:

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

Code #2:

applet




# 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])>

>

>

Uitgang:

[[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])>