☠ 7. feladat¶
Kulcsszavas függvény változó számú argumentummal¶
Írjunk egy függvényt, amely egy adott x array-ben tárólt valós értékekre kiértékel egy tetszőleges polinomfüggvényt, vagy ábrázolja azt!
- A polinom-együtthatókat egy tetszőleges hosszúságú args nevű váltózóban kapjuk.
- Ha a kwargs kulcsszavas argumentumban a
csinal_kepetkulcsszoTrueakkor ábrázoljuk a polinomot és ne térjünk vissza semmilyen értékkel, ellenkező esetben a függvény térjen vissza egy array változóval amely polinom értékeit tartalmazza az x válltozóban tárolt helyeken. - Az ábrázolás stilisztikai paramétereit a függvénynek adott egyéb kulcsszavas argumok segítségével állítsuk be!
Megoldás¶
Mivel ebben a feladatban a pylab több funkcióját használni fogjuk, először ezt töltsük be!
In [3]:
%pylab inline
Ezután írjuk meg a függvényt! Amit mindenkképpen meg kell csinálni, hogy az adott x értékekhez kiszámoljuk az y értékeket. Ezt egy for-ciklussal tudjuk megtenni. Úgy vesszük, hogy a paraméterként kapott tömbben az i. érték az i hatvány együtthatója. Ezután a kwargs-t vizsgáljuk meg, hogy benne van-e a kérdéses kulcsszó:
In [41]:
def fuggveny(x,*args,**kwargs):
"Kiértékelek vagy ábrázolok egy polynom-ot"
y = zeros(len(x)) # Ebbe a tömbbe fogjuk beírni a függvényértékeket
for i in range(0, len(args)): # Végig megyünk a tömb elemein, és kiszámítjuk a hozzájuk tartozó függvényértéket
y += args[i]*(x**i)
if "csinal_kepet" in kwargs and kwargs["csinal_kepet"]: # Ha meg van adva a kérdéses paraméter, ábrázolja
del kwargs["csinal_kepet"]
plot(x, y, **kwargs)
else: # Ha nincs, csak visszaadjuk eredményként
return y
Ezután kipróbáljuk néhány próbafüggvénnyel.
In [44]:
fuggveny(arange(0,1,0.05), 1,0,1, csinal_kepet=True, color="green", marker="o")
In [45]:
fuggveny(arange(0,100,2), 1,2,3,4,5)
Out[45]: