在notebook环境中运行时,会有一个内置的__IPYTHON__
变量:
一般为True
值。
print(__IPYTHON__)
输出
True
而如果这段代码直接在*.py
文件中执行,则会报错。
因此可以通过判断是否存在内置的__IPYTHON__
变量来判断是否是notebook环境。
即
hasattr(__builtins__,"__IPYTHON__")
例如某些库在不同的环境中有不同的调用方法,例如:
if hasattr(__builtins__,"__IPYTHON__"):
from tqdm.notebook import tqdm
print("notebook环境")
else:
from tqdm import tqdm
print("普通运行环境")