Как отправить вывод% времени в файл в Jupyter/IPython Notebook?

Question

Волшебная команда %time генерирует время, которое требуется для выполнения заданного сегмента кода (оператора) в выходной ячейке в Jupyter Notebook. Я хотел бы, чтобы это было добавлено к указанному файлу.

Моя цель - найти время выполнения алгоритма в разных настройках. Я могу автоматизировать параметры алгоритма, хотя не могу сохранить выходные данные %time (а также %timeit) Волшебная команда в файле для дальнейшей обработки.

Это возможно?

Answer 1

Репостить как ответ:

%timeit (но нет %time) имеет -o опция, которая заставляет его возвращать объект, так что вы можете сделать:

res = %timeit -o foo()

Затем вы можете получить номера из res и записать их в файл, как пожелаете.

Для большей гибкости используйте модуль timeit в стандартной библиотеке.