logo

numpy.where() Pythonissa

The numpy.where() -funktio palauttaa syötetaulukon elementtien indeksit, joissa annettu ehto täyttyy.

Syntaksi : numpy.where(ehto[, x, y])
Parametrit:
kunto: Kun tosi, tuotto x, muuten tuotto y.
x, y: Arvot, joista valita. x:n, y:n ja ehdon on oltava lähetettävissä johonkin muotoon.

Palautukset:
ulos: [ndarray tai tuple of ndarrays] Jos sekä x että y on määritetty, tulostaulukko sisältää elementtejä x:stä, jossa ehto on True, ja elementtejä y:stä muualta.



Jos vain ehto on annettu, palauta monikko condition.nonzero(), indeksit, joissa ehto on True.

Koodi #1:


onnea



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

>

>

kokoinen lateksifontti

Lähtö:

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

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

Näyttelijä Rakul Preet Singh

>

>

Lähtö:

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