假设我在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的部分。但是我无法真正将其推到可行的程度。
是否存在任何库或现成的功能来实现这一目标?
谢谢