Python tqdm 导入检查 jupyter notebook 或 lab 是否正在运行

Question

我有一个模块 (tqdm)，我需要根据我是在 jupyter notebook 还是 jupyter lab 环境中运行 .ipynb 来以不同的方式导入它。有没有办法我可以在 python 中确定这一点？例如：

if <jupyter notebook>:
    from tqdm import tqdm_notebook as tqdm 
elif <jupyter lab>:
    from tqdm import tqdm
else:
    print("Not in jupyter environment.")

Answer 1

# either:
from tqdm.autonotebook import tqdm
# or to suppress the warning:
from tqdm.auto import tqdm

对于其他模块/检查，请参阅How can I check if code is executed in the IPython notebook?（但请注意，尽管不受欢迎，但已接受的答案是“这不是有意设计的。”）