从数组中提取所有最小值:

时间:2019-05-27 12:17:35

标签: python arrays numpy

假设我在numpy中有一个数组,其内容为:

arr = np.array([[1,2,7,6],[2,3,2,1],[10,8,1,3],[9,7,4,2], [np.nan, np.nan, np.nan, np.nan]])

我想提取周围条目比较好的所有条目。一种“局部最小值”,这样我可以获得以下索引:

indices = [(0,0), [(2,2), (2,3)]]

请注意,列表中的第二项是元组列表。我希望这种算法能够识别局部最小区域以及局部最小点。 NAN值应简单地被忽略,并被视为“边界”。

我尝试使用np.gradient,因为我认为它应该允许我提取导数为0的部分。但是我无法真正将其推到可行的程度。

是否存在任何库或现成的功能来实现这一目标?

谢谢

0 个答案:

没有答案