Python数组减法循环回到高数而不是给出负值

Question

这是我的代码: 我正在使用 numpy 和 opencv

q = np.array(image)
q = q.reshape(-1, q.shape[2])
r = np.subtract(q,p)
print r

基本上，如果我的 q 数组中的值大于 p，减法循环回到 256 并减去那里剩下的内容。如果减法变为负数，我宁愿得到 0 的值。有人知道这样做的好方法吗？

Answer 1

您可以更改为支持负整数的 int16 并将负值设置为 0，您的值正在换行，因为您有 uint8:

arr1 = np.array([100, 200, 255],dtype=np.int16)
arr2 = np.array([180, 210, 100],dtype=np.int16)

sub_arr = np.subtract(arr1, arr2)
sub_arr[sub_arr < 0] = 0
print(sub_arr)
[  0   0 155]

要更改数组，您可以使用 array.astype(np.int16) 将 uint8 更改为 np.int16 并使用同样在减去之后再变回来。

arr1 = np.array([100, 200, 255],dtype=np.uint8)
arr2 = np.array([180, 210, 100],dtype=np.uint8)
_arr2 = arr2.astype(np.int16)
sub_arr = np.subtract(arr1, _arr2)

sub_arr[sub_arr < 0] = 0
sub_arr = sub_arr.astype(np.uint8)
print(sub_arr)

或者也可以使用 np.clip :

arr1 = np.array([100, 200, 255],dtype=np.uint8)
arr2 = np.array([180, 210, 100],dtype=np.uint8)

sub_arr = np.subtract(arr1, arr2.astype(np.int16)).clip(0, 255).astype(np.uint8)
print(sub_arr)
[  0   0 155]