VS Code中如何为Jupyter Notebook加载项目环境变量文件？

完全可以通过settings.json中的jupyter.runStartupCommands配置来实现需求，VS Code会自动解析其中的${workspaceFolder}变量，无需修改Notebook文件，也不用额外安装包。

具体配置步骤 #

1. 打开VS Code的设置界面（快捷键Ctrl+,），搜索并找到Jupyter: Run Startup Commands选项
2. 点击编辑 in settings.json，添加以下配置：

"jupyter.runStartupCommands": [
    // 注入工作区路径到环境变量
    "import os; os.environ['WORKSPACE_DIR'] = '${workspaceFolder}'",
    // 读取并加载项目环境变量文件
    "import os; from pathlib import Path; env_path = Path(os.environ['WORKSPACE_DIR']) / '.vscode/env'; if env_path.exists(): with open(env_path, 'r') as f: [os.environ[k.strip()] = v.strip() for k, v in (line.split('=', 1) for line in f if line.strip() and not line.strip().startswith('#'))]"
]

说明 #

• 这段代码会在Jupyter内核启动时自动执行：
  1. 先将当前工作区的绝对路径存入WORKSPACE_DIR环境变量
  2. 定位到项目下的.vscode/env文件，逐行读取有效配置（跳过空行和注释行），将键值对添加到系统环境变量中
• 要求.vscode/env文件格式为每行一个环境变量，示例：

  DB_HOST=localhost
  DB_PORT=5432
  # 注释行会被忽略
  API_KEY=abc123
  

内容的提问来源于stack exchange，提问作者guest