Obvezna naloga

Ideja te naloge je, da jo rešite z eno samo preprosto vrstico v numpyju. Nobenega for ali while, nobenih izpeljanih seznamov ali česa podobnega!

Pravilno napisna funkcija bo dolga eno vrstico, ki bo krajša od vrstice

def srecne_drzave(teze, vrednosti, drzave, meja):

Tore: napišite funkcijo srecne_drzave(teze, vrednosti, drzave, meja).

• teze vsebuje težo uvoženih banan,
• vrednosti pove, koliko je posamezna država plačala za te banane,
• drzave pa so imena držav. Vse tri tabele so enake dolžine in vsebujejo podatke v enakem vrstnem redu.

Gre, očitno, za to, kar vračata prvi dve funkciji iz prejšnje domače naloge.

Funkcija naj vrne tabelo z imeni držav, ki so za svoje banane plačale manj od podane meje. Države morajo v tabeli nastopati v enakem vrstnem redu kot v podani tabeli drzave. Z drugimi besedami, tabela, ki jo vrnete, mora biti takšna kot tabela drzave, vendar le s tistimi državami, ki so plačale manj kot meja.

Pomoč: s tabelo nizov lahko delamo enake stvari, kot smo jih na predavanjih s tabelami števil - tudi tabelo držav lahko indeksiramo s tabelami indeksov, s tabelo True-jev in False-ov ipd.

Dodatna naloga

Ta je tako, za zraven. Neobvezno.

Dam vam funkcijo

def indeks_drzave(drzave, drzava):
    return np.where(drzave == drzava)[0][0]

ki vrne indeks države v tabeli držav.

Napišite funkcijo vec_kot(teze, drzave, drzava), ki prejme tabelo s težami, tabelo z državami in ime neke države. Vrne število držav, ki so kupile več banan kot podana država.

Testi