Jupyter Notebook间调用传参及变量共享失败求助

一、%run魔法命令传参失效 #

问题表现 #

用%run "notebook2.ipynb" 1 2调用Notebook时，目标文件里sys.argv拿不到传入的1和2，输出只有内核启动的相关参数。

解决办法 #

%run的参数不能直接通过sys.argv获取，换这两种方式：

• 方式1：通过get_ipython()读取参数
  在被调用的notebook2.ipynb里加这段代码：

params = get_ipython().user_ns.get('sys.argv', [])[2:]
print(params)  # 会输出 ['1', '2']

• 方式2：用-i参数注入变量
  先在notebook1.ipynb里定义变量，再用-i参数让被调用的Notebook直接使用这些变量：

# notebook1.ipynb
arg1 = 1
arg2 = 2
%run -i "notebook2.ipynb"

# notebook2.ipynb
print(arg1, arg2)  # 直接输出 1 2

二、%store共享变量报错TypeError: 'PickleShareDB' object is not subscriptable #

问题表现 #

用%store存了变量后，另一个Notebook用%store -r读取时触发这个错误。

解决办法 #

这个错误大多是存储路径权限或版本兼容问题，试试这几个方案：

• 重置存储缓存
  在任意Notebook里执行%store -z清空所有已存变量，重新存储读取：

# notebook1.ipynb
%store -z  # 清空所有存储的变量
%store df
%store var

# notebook2.ipynb
%store -r df
%store -r var

• 手动指定存储目录
  给PickleShareDB指定一个有读写权限的本地目录，两个Notebook都要执行这段：

from ipykernel.pickleutil import PickleShareDB
import os
# 用当前目录下的jupyter_store文件夹存数据
db_path = os.path.join(os.getcwd(), 'jupyter_store')
os.makedirs(db_path, exist_ok=True)
get_ipython().user_ns['store'] = PickleShareDB(db_path)

之后再正常用%store和%store -r操作变量。

• 调整依赖版本
  如果是版本冲突，试试降级pickleshare：

pip install pickleshare==0.7.5

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