python - 打印 joblib 中函数的输出

Question

我正在尝试使用 joblib 来并行化一个在函数上运行的循环。我希望显示函数的中间 print 命令，而不仅仅是函数的 return 值。

from joblib import Parallel, delayed

def dummy(i):
    print("the value passed is",i)

Parallel(n_jobs=2)(delayed(dummy)(i) for i in range(0,5,1))

我得到以下输出:

[None,
 None,
 None,
 None,
 None]

我想得到以下输出(或类似的东西):

[the value passed is 0,
the value passed is 1,
the value passed is 2,
the value passed is 3,
the value passed is 4]

编辑: 我只是注意到它确实在启动我的 Jupyter 笔记本的终端窗口中打印了我需要的输出。关于如何在我的笔记本中实际打印它的任何想法。 提前致谢。

Answer 1

from joblib import Parallel, delayed

def dummy(i):
    return "the value passed is " + str(i)

Parallel(n_jobs=2)(delayed(dummy)(i) for i in range(0,5,1))

输出::

[the value passed is 0,
the value passed is 1,
the value passed is 2,
the value passed is 3,
the value passed is 4]