python - 如何使用边距进行 numpy Logical_and 操作？

Question

我有两个 numpy 向量数组，一个包含二进制值，即 1 或 0，另一个包含浮点值，即 0 到 1 之间的任何值。

我想使用numpy.logic_and运算符，并让它在二进制值在浮点正负0.2范围内时返回true。因此，即 float 0.1 将返回 true，0.4 返回 false。

我该如何解决这个问题？

Answer 1

我认为你想要的是np.isclose。在这种情况下，实现将是:

bin_arr = np.random.randint(2, size = 100)
float_arr = np.random.rand(100)
out = np.isclose(bin_arr.astype(float), float_arr, atol = .2)

请注意，虽然 logic_and 是 ufunc (Universal Function)具有扩展功能，np.isclose 则不然。