Python - 计算渐变

Question

如果我有一些函数作为数组，我想计算它的渐变，那么我将使用渐变函数：

from numpy import gradient
g = gradient(f)

然后我得到那个渐变，但如果我想让渐变等于零，我怎样才能找到极值点？

通常我不是程序员，而是使用Python。我不明白如何编写这样的函数。这是我的物理项目。我明白，如果我只是写出渐变等于零，我将保存零值。但是当我尝试编写该函数（百万次）时，我总是失败，主要是语法问题。我是对的，请问'if'功能好吗？有没有人有一些例子？感谢。

Answer 1

我不确定我是否理解你的问题，但如果你需要渐变为零的位置，你可以这样写：

from numpy import gradient

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