python - Python で numpy unsigned int の差を計算するより良い方法

Question

numpy.uint8を使用して Python で 2 つの の絶対差を計算しています。が より大きいd = abs(a - b)場合、オーバーフロー エラーが発生します。これを回避する最善の方法は何ですか?ba

Answer 1

あなたのコメントが示すように、それらはints ではありません。彼らはnumpy.uint8s です。intそれらをsに変換するだけです:

>>> a, b = map(numpy.uint8, (50, 60))
>>> a - b
__main__:1: RuntimeWarning: overflow encountered in ubyte_scalars
246
>>> a, b = map(int, (a, b))
>>> a - b
-10

速度が気になるので、ここにいくつかのテストを示します (感謝の気持ちを込めてSvenのフォームを借ります)。

>>> %timeit abs(int(a) - int(b))
1000000 loops, best of 3: 410 ns per loop
>>> %timeit a - b if a > b else b - a
1000000 loops, best of 3: 470 ns per loop

そうです、その方が高速ですが、何億回も実行することについて話しているのでない限り、それは少しも問題になりません.

Answer 2

最も簡単な方法は、最初に手動で数値を Python の int に変換することです。

d = abs(int(a) - int(b))

Python int はオーバーフローできません (メモリがいっぱいでない限り)。

Answer 3

私にとっては、オーバーフローエラーは発生しませbんa。境界内にとどまるには、uint8次の関数を使用します。

def abs_dev (a, b) :
    sub1 = a - b
    sub2 = b - a
    mask = a < b
    sub1[mask] = sub2[mask]
    return sub1