python - 检查 numpy 数组中的所有元素是否与数字匹配

Question

在我的 python 测试脚本中，我想断言 numpy 数组的所有元素是否非常接近 1.0 或等于 0.0。该数组如下所示:

[[0.9999999991268851 1.0000000223517418 0.999999986961484 ...,
  0.9999999841675162 1.0000000074505806 0.9999999841675162]
 [0.9999999991268851 1.0000000223517418 0.999999986961484 ...,
  0.9999999841675162 1.0000000074505806 0.9999999841675162]
 [0.9999999991268851 1.0000000223517418 0.999999986961484 ...,
  0.9999999841675162 1.0000000074505806 0.9999999841675162]
 ..., 
 [1.0000000198488124 1.0000000074505806 1.000000002568413 ...,
  0.9999999888241291 0.9999999925494194 0.0]
 [1.000000011001248 0.9999999850988388 0.9999999869323801 ...,
  1.0000000186264515 0.9999999925494194 0.0]
 [1.000000011001248 0.9999999850988388 0.9999999869323801 ...,
  1.0000000186264515 0.9999999925494194 0.0]]

我想过使用 numpy.allclose 或 numpy.array_equal，但这两者都没有意义。理想情况下，该函数应该能够在测试场景中使用

Answer 1

您可以获取 0 个元素并使用 bool 索引将它们屏蔽掉。完成后，np.allclose正是您想要的:

zeros = arr == 0.0
without_zeros = arr[~zeros]
np.allclose(without_zeros, 1, ...)