4

我有一个 (960,960) 数组，我试图找到临界点，以便找到局部极值。

我曾尝试使用 np.diff 和 np.gradient，但遇到了一些麻烦，我不确定要使用哪个函数。

np.diff 提供计算二阶差异的选项，但梯度不提供。

我应该如何获得关键点？

我试过了

diff = np.diff(storm, n=2)                    

dxx = diff[0]                                                                                                                                  
dyy = diff[1]                                                                                                                                  

derivative = dyy/dxx

我在这里遇到了问题，因为 dxx 上的一些值等于零。

然后是选项

gradient = np.gradient(storm)
g2 = np.gradient(gradient)

但这会给我想要的东西吗？

4

1 回答 1

5

临界点是函数的一阶导数（或多维情况下的梯度）为 0 的点。因此，您应该检查函数的 x 和 y 差异。numpy的diff功能适用于这种情况。

因此，如果 x-y- 方向上两个相邻元素之间的差异接近于 0，则可以说该点为临界点。假设您的函数是平滑的，那么差异就会改变其符号（从负变为正，反之亦然）。

# get difference in x- and y- direction
sec_grad_x = np.diff(storm,n=1,axis=0)
sec_grad_y = np.diff(storm,n=1,axis=1)

cp = []
# starts from 1 because diff function gives a forward difference
for i in range(1,n-1):
    for j in range(1,n-1):
        # check when the difference changes its sign
        if ((sec_grad_x[i-1,j]<0) != (sec_grad_x[i-1+1,j]<0)) and \
           ((sec_grad_y[i,j-1]<0) != (sec_grad_y[i,j-1+1]<0)):
            cp.append([i,j,  storm[i,j]])

cp = np.array(cp)
于 2014-05-08T01:02:18.353 回答