Python - Berechnen eines Gradienten

Question

Wenn ich eine Funktion als Array haben, und ich will seine Steigung berechnen dann werde ich die Steigung-Funktion:

from numpy import gradient 
g = gradient(f) 

Dann bekomme ich die Steigung, aber Wenn ich will Lass die Steigung gleich Null sein, wie finde ich die Extrempunkte?

Normalerweise bin ich kein Programmierer und ich benutze Python. Ich verstehe noch nicht, wie ich eine Funktion schreiben soll. Es ist für mein Physikprojekt. Ich verstehe, dass, wenn ich nur schreibe, dass der Gradient gleich Null ist, ich nur den Wert Null speichern werde. Aber als ich versuchte, diese Funktion (Millionen mal) zu schreiben, habe ich immer versagt, meistens Syntaxproblem. Habe ich recht, es wird bitte "wenn" funktionieren? Hat jemand ein Beispiel? Vielen Dank.

Answer 1

Ich bin mir nicht sicher, ob ich verstehe Ihre Frage, aber wenn man die Positionen müssen, wo der Gradient Null ist, kann man so etwas schreiben:

from numpy import gradient 

g = gradient(f) 
zeros = [ i for i,x in enumerate(g) if x.any() == 0 ] 
print(zeros)