Numpy uint16 加法无溢出异常

Question

I want to add noise to my uint16 signal, but I want to avoid overflows in case some signal values are zero:

np.seterr(all='raise')
noise = np.random.normal(0,2,9*9).reshape(9,9)
noise = np.around(noise,out=noise)
noise = noise.astype(np.uint16,copy=True)
signal = np.zeros((9,9), dtype=np.uint16)
try:
    signal += noise
except FloatingPointError:
    print("Overflow Detected")
    # Handle overflows
print(test)

Unfortunately there is no exception raised. The example from the doc works.

>>> np.int16(32000) * np.int16(3)
---------------------------------------------------------------------------
FloatingPointError                        Traceback (most recent call last)
<ipython-input-84-5e468485740e> in <module>
----> 1 np.int16(32000) * np.int16(3)

Is there a way to enable an exception for uint16 addition?

Or is there an efficient way to check for overflows which does not involve going through the array element-by-eliment?

Answer 1

Numpy supports broadcasting comparison operators. Thus, the problem can be solved by checking the result for illegal values. In my case the signal is actually 12-bit, so any values above 2**12-1 must have come from an overflow.

noise = np.random.normal(0,2,9*9).reshape(9,9)
noise = np.around(noise,out=noise)
noise = noise.astype(np.uint16,copy=True)
test = np.zeros((9,9), dtype=np.uint16)
test += noise
check = test > 2**12-1
if check.any():
    print("Overflow detected")

If you have actual 16-bit values, the arithmetic operation (eg addition) can be performed in a 32-bit array; You can then check for illegal values in the same fashion.

You can also handle the overflow immediately with np.where() :

np.where(a > OVERFLOW_BOUNDARY, f(a), a)

Note that you probably want to differentiate between an overflow and an underflow in your handler function.