web-dev-qa-db-ja.com

Python:*と**が/とsqrt()よりも速いのはなぜですか?

コードを最適化しているときに、次のことに気づきました。

>>> from timeit import Timer as T
>>> T(lambda : 1234567890 / 4.0).repeat()
[0.22256922721862793, 0.20560789108276367, 0.20530295372009277]
>>> from __future__ import division
>>> T(lambda : 1234567890 / 4).repeat()
[0.14969301223754883, 0.14155197143554688, 0.14141488075256348]
>>> T(lambda : 1234567890 * 0.25).repeat()
[0.13619112968444824, 0.1281130313873291, 0.12830305099487305]

そしてまた:

>>> from math import sqrt
>>> T(lambda : sqrt(1234567890)).repeat()
[0.2597470283508301, 0.2498021125793457, 0.24994492530822754]
>>> T(lambda : 1234567890 ** 0.5).repeat()
[0.15409398078918457, 0.14059877395629883, 0.14049601554870605]

pythonがCで実装されている方法と関係があると思いますが、なぜそうなのか誰かが説明したいと思うでしょうか?

80
mac

結果の(やや予想外の)理由は、Pythonは浮動小数点の乗算とべき乗を含む定数式を折りたたむようですが、除算は含まないようです。math.sqrt()はまったく別の獣です。そのためのバイトコードがなく、関数呼び出しが含まれるためです。

Python 2.6.5では、次のコード:

_x1 = 1234567890.0 / 4.0
x2 = 1234567890.0 * 0.25
x3 = 1234567890.0 ** 0.5
x4 = math.sqrt(1234567890.0)
_

次のバイトコードにコンパイルされます。

_  # x1 = 1234567890.0 / 4.0
  4           0 LOAD_CONST               1 (1234567890.0)
              3 LOAD_CONST               2 (4.0)
              6 BINARY_DIVIDE       
              7 STORE_FAST               0 (x1)

  # x2 = 1234567890.0 * 0.25
  5          10 LOAD_CONST               5 (308641972.5)
             13 STORE_FAST               1 (x2)

  # x3 = 1234567890.0 ** 0.5
  6          16 LOAD_CONST               6 (35136.418286444619)
             19 STORE_FAST               2 (x3)

  # x4 = math.sqrt(1234567890.0)
  7          22 LOAD_GLOBAL              0 (math)
             25 LOAD_ATTR                1 (sqrt)
             28 LOAD_CONST               1 (1234567890.0)
             31 CALL_FUNCTION            1
             34 STORE_FAST               3 (x4)
_

ご覧のとおり、乗算とべき乗はコードのコンパイル時に行われるため、まったく時間がかかりません。分割は実行時に発生するため、時間がかかります。平方根は、4つの中で最も計算コストの高い操作であるだけでなく、他の操作では発生しないさまざまなオーバーヘッド(属性ルックアップ、関数呼び出しなど)も発生します。

定数畳み込みの影響を排除すると、乗算と除算を分離することはほとんどありません。

_In [16]: x = 1234567890.0

In [17]: %timeit x / 4.0
10000000 loops, best of 3: 87.8 ns per loop

In [18]: %timeit x * 0.25
10000000 loops, best of 3: 91.6 ns per loop
_

math.sqrt(x)は、実際には_x ** 0.5_よりも少し高速です。これは、おそらく後者の特殊なケースであるため、オーバーヘッドにもかかわらず、より効率的に実行できるためです。

_In [19]: %timeit x ** 0.5
1000000 loops, best of 3: 211 ns per loop

In [20]: %timeit math.sqrt(x)
10000000 loops, best of 3: 181 ns per loop
_

edit 2011-11-16:定数式の折りたたみは、Pythonののぞき穴オプティマイザーによって行われます。ソースコード(_peephole.c_)には、定数除算が折りたたまれない理由を説明する次のコメントが含まれています。

_    case BINARY_DIVIDE:
        /* Cannot fold this operation statically since
           the result can depend on the run-time presence
           of the -Qnew flag */
        return 0;
_

_-Qnew_フラグは、 PEP 238 で定義されている「真の除算」を有効にします。

113
NPE