NumPy-moduuli tarjoaa funktion numpy.where() elementtien valitsemiseksi ehdon perusteella. Se palauttaa elementit, jotka on valittu joukosta a tai b ehtojen mukaan.
Jos esimerkiksi kaikki argumentit -> ehto, a & b välitetään parametrissa numpy.where(), se palauttaa elementit, jotka on valittu joukosta a & b riippuen ehdon tuottamista arvoista bool-taulukossa.
onnea
Jos vain ehto on annettu, tämä funktio on lyhenne funktiolle np.asarray (condition).nonzero(). Vaikka nollasta poikkeava olisi suositeltava suoraan, koska se käyttäytyy oikein alaluokissa.
Syntaksi:
numpy.where(condition[, x, y])
Parametrit:
Nämä ovat seuraavat parametrit numpy.where()-funktiossa:
ehto: array_like, bool
Jos tämän parametrin arvoksi on asetettu True, tuotto x, muuten tuotto y.
x, y: array_like:
kokoinen lateksifontti
Tämä parametri määrittää arvot, joista valita. X:n, y:n ja ehdon on oltava lähetettävissä jossain muodossa.
Palautukset:
Tämä funktio palauttaa taulukon elementeillä x:stä, jossa ehto on True, ja y:n elementtejä muualta.
Esimerkki 1: np.where()
import numpy as np a=np.arange(12) b=np.where(a<6,a,5*a) b < pre> <p> <strong>In the above code</strong> </p> <ul> <li>We have imported numpy with alias name np.</li> <li>We have created an array 'a' using np.arange() function.</li> <li>We have declared the variable 'b' and assigned the returned value of np.where() function.</li> <li>We have passed the array 'a' in the function.</li> <li>Lastly, we tried to print the value of b.</li> </ul> <p>In the output, the values ranging from 0 to 5 remain the same as per the condition, and the other values have been multiplied with 5.</p> <p> <strong>Output:</strong> </p> <pre> array([ 0, 1, 2, 3, 4, 5, 30, 35, 40, 45, 50, 55]) </pre> <h3>Example 2: For multidimensional array</h3> <pre> import numpy as np a=np.arange(12) b=np.where([[True, False], [True, True]],[[1, 2], [3, 4]],[[9, 8], [7, 6]]) b </pre> <p> <strong>Output:</strong> </p> <pre> array([[1, 8], [3, 4]]) </pre> <h3>Example 3: Broadcasting x, y, and condition</h3> <pre> import numpy as np x, y = np.ogrid[:3, :4] a=np.where(x > y, x, 10 + y) a </pre> <p> <strong>Output:</strong> </p> <pre> array([[10, 11, 12, 13], [ 1, 11, 12, 13], [ 2, 2, 12, 13]]) </pre> <p> <strong>In the above code</strong> </p> <ul> <li>We have imported numpy with alias name np.</li> <li>We have created an array 'a' using np.arange() function. </li> <li>We declared the variable 'b' and assigned the returned value of np.where() function.</li> <li>We have passed a multidimensional array of boolean as a condition and x and y as an integer arrays.</li> <li>Lastly, we tried to print the value of b.</li> </ul> <p>In the output, the x value has been compared to y value if it satisfied the condition, then it will be printed x value otherwise, it will print y value, which has passed as an argument in the where() function.</p> <h3>Example 4: Broadcasting specific value</h3> <pre> x=np.array([[0,1,2],[0,2,5],[0,4,8]]) y=np.where(x<4,x,-2) y < pre> <p> <strong>Output:</strong> </p> <pre> array([[ 0, 1, 2], [ 0, 2, -2], [ 0, -2, -2]]) </pre> <hr></4,x,-2)></pre></6,a,5*a)>
Esimerkki 2: Moniulotteiselle taulukolle
import numpy as np a=np.arange(12) b=np.where([[True, False], [True, True]],[[1, 2], [3, 4]],[[9, 8], [7, 6]]) b
Lähtö:
Näyttelijä Rakul Preet Singh
array([[1, 8], [3, 4]])
Esimerkki 3: Lähetetään x, y ja ehto
import numpy as np x, y = np.ogrid[:3, :4] a=np.where(x > y, x, 10 + y) a
Lähtö:
array([[10, 11, 12, 13], [ 1, 11, 12, 13], [ 2, 2, 12, 13]])
Yllä olevassa koodissa
- Olemme tuoneet numpyn aliasnimellä np.
- Olemme luoneet taulukon 'a' käyttämällä np.arange()-funktiota.
- Ilmoitimme muuttujan 'b' ja määritimme palautetun arvon funktiolle np.where().
- Olemme hyväksyneet moniulotteisen boolen taulukon ehdoksi ja x ja y kokonaislukutaulukoiksi.
- Lopuksi yritimme tulostaa b:n arvon.
Tulosteessa x-arvoa on verrattu y-arvoon, jos se täytti ehdon, sitten tulostetaan x-arvo muuten tulostetaan y-arvo, joka on mennyt argumenttina missä()-funktiossa.
Esimerkki 4: Lähetyskohtainen arvo
x=np.array([[0,1,2],[0,2,5],[0,4,8]]) y=np.where(x<4,x,-2) y < pre> <p> <strong>Output:</strong> </p> <pre> array([[ 0, 1, 2], [ 0, 2, -2], [ 0, -2, -2]]) </pre> <hr></4,x,-2)>