📜  运行项目文件夹中的所有 jupyter 笔记本 - Python (1)

📅  最后修改于: 2023-12-03 15:41:55.017000             🧑  作者: Mango

运行项目文件夹中的所有 jupyter 笔记本 - Python

如果您经常使用Jupyter笔记本进行项目开发,您可能有时候需要同时运行多个笔记本文件来进行数据分析或模型预测。这种情况下,手动一个一个运行笔记本文件会非常耗时费力。为了提高工作效率,我们可以编写一个Python脚本,来快速运行项目文件夹中的所有Jupyter笔记本文件。

思路

我们可以通过Python中的os模块,来遍历项目文件夹中的所有Jupyter笔记本文件,并通过jupyter-notebook命令来启动它们。具体思路如下:

  1. 使用os模块获取当前文件夹路径,并指定Jupyter笔记本文件所在的子文件夹路径。
  2. 使用os模块遍历子文件夹中所有的Jupyter笔记本文件。
  3. 使用subprocess模块,通过jupyter-notebook命令来启动每一个Jupyter笔记本文件。
代码
import os
import subprocess

def run_jupyter_notebooks():
    # 获取当前文件夹路径
    current_dir = os.getcwd()
    # 指定Jupyter笔记本文件所在的子文件夹路径
    notebooks_dir = os.path.join(current_dir, 'notebooks')
    
    # 遍历子文件夹中所有的Jupyter笔记本文件
    for root, dirs, files in os.walk(notebooks_dir):
        for file in files:
            if file.endswith('.ipynb'):
                notebook_file = os.path.join(root, file)
                # 使用subprocess模块,通过jupyter-notebook命令来启动Jupyter笔记本文件
                subprocess.run([f'jupyter-notebook {notebook_file}'], shell=True)

# 调用函数运行所有Jupyter笔记本文件
run_jupyter_notebooks()
结论

run_jupyter_notebooks()函数会自动遍历项目文件夹中的所有Jupyter笔记本文件,并逐个启动它们。我们只需要运行这个函数,就能自动运行所有的Jupyter笔记本文件,避免了手动运行笔记本的繁琐操作。