% timeit и переназначение переменной

Question

Удивительная магия ipython %timeit ошибка:

In[1]: a = 2

In[2]: %timeit a = 2 * a

Traceback (most recent call last):
  File "...\site-packages\IPython\core\interactiveshell.py", line 3326, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-97-6f70919654d1>", line 1, in <module>
    get_ipython().run_line_magic('timeit', 'a = 2 * a')
  File "...\site-packages\IPython\core\interactiveshell.py", line 2314, in run_line_magic
    result = fn(*args, **kwargs)
  File "<...\site-packages\decorator.py:decorator-gen-61>", line 2, in timeit
  File "...\site-packages\IPython\core\magic.py", line 187, in <lambda>
    call = lambda f, *a, **k: f(*a, **k)
  File "...\site-packages\IPython\core\magics\execution.py", line 1158, in timeit
    time_number = timer.timeit(number)
  File "...\site-packages\IPython\core\magics\execution.py", line 169, in timeit
    timing = self.inner(it, self.timer)
  File "<magic-timeit>", line 1, in inner
UnboundLocalError: local variable 'a' referenced before assignment

Так что %timeit не любит самостоятельное переназначение. Почему? В любом случае, чтобы преодолеть это?

Answer 1

Как и в базовом модуле timeit, оператор timed интегрирован в сгенерированную функцию, которая выполняет синхронизацию. Присвоение a заставляет функцию иметь локальную переменную a, скрывающую глобальную переменную. Это та же проблема, как если бы вы сделали

a = 2

def f():
    a = 2 * a

f()

, хотя сгенерированная функция имеет больше кода, чем эта.

Answer 2

Время работы ячейки с инициализацией:

In [142]: %%timeit a=2 
     ...: a = 2*a 
     ...:  
     ...:                                                                       
2.68 µs ± 66.7 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

Answer 3

Вместо этого вы можете использовать магию %%, чтобы избежать ошибки присваивания:

%%timeit
a = 2
a = a*2