python - 在 2d numpy 数组的每一行中找到最小的非零值

Question

这个问题在这里已经有了答案:





Find the min/max excluding zeros in a numpy array (or a tuple) in python

(6 个回答)


去年关闭。




我试图在 2d numpy 数组的每一行中找到最小的非零值，但还没有找到一个优雅的解决方案。我看过其他一些帖子，但没有一个解决完全相同的问题，例如
Minimum value in 2d array或 Min/Max excluding zeros but in 1d array .
例如对于给定的数组:

x = np.array([[3., 2., 0., 1., 6.], [8., 4., 5., 0., 6.], [0., 7., 2., 5., 0.]])

答案是:

[1., 4., 2.]

Answer 1

一种方法是将零重新分配给 np.inf，然后每行取 min:

np.where(x>0, x, np.inf).min(axis=1)

输出:

array([1., 4., 2.])