%timeit 單獨工作但失敗與 .py 文件

Question

我正在嘗試將 majic (%timeit) 命令結果捕獲到文本文件，同樣在 ipython 中單獨執行但在 .py 文件中調用，它拋出語法錯誤。 請幫助我我的腳本中到底缺少什么。

from scripts.download_sftp_extract_fun import sftp_extract
import sys
import timeit
times = %timeit -o sftp_extract.download_sftp_extract(extract_name="RE_AMS_SUBD_SLS_WKLY_")
print(times,file=open("logs/log_re_ams_subd_sls_wkly.txt",'w'))

錯誤信息：-

In [26]: exec(open('scripts/extract_download_re_ams_subd_sls_wkly_.py').read())
Traceback (most recent call last):

  File "/usr/local/lib/python3.8/dist-packages/IPython/core/interactiveshell.py", line 3417, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)

  File "<ipython-input-26-21b570da7ee2>", line 1, in <module>
    exec(open('scripts/extract_download_re_ams_subd_sls_wkly_.py').read())

  File "<string>", line 4
    times=%timeit -o sftp_extract.download_sftp_extract(extract_name="RE_AMS_SUBD_SLS_WKLY_")
          ^
SyntaxError: invalid syntax

Answer 1

timeit的正確用法如下：

import timeit
timeit.timeit("sftp_extract.download_sftp_extract(extract_name=\"RE_AMS_SUBD_SLS_WKLY_\")")

查看它的文檔以獲取更多詳細信息