python - 如何识别 numpy 数组中的最小值，不包括对角线零？

Question

    >>> x
    array([[ 0,  2,  3,  4],
           [ 6,  0,  8,  9],
           [ 1,  2,  0, -9]
           [ -9, 4,  3,  0])

我希望能够将 -9 识别为 min(x)，而不是 0。

谢谢

Answer 1

1. 将对角线设置为其 的最大值dtype。对于浮点类型，那将是np.inf，但对于整数，您必须更加努力。

   x[np.diag_indices_from(x)] = np.iinfo(x.dtype).max
   

2. 采取min.

3. 将对角线设置回零。

Answer 2

Since -9 < 0 x.min() should work without any trickery. You only need to worry if you min won't be negative.