我必须将带有值的生成器作为参数传递给特定函数。
function Func 输入 pd.DataFrame 和一个值。
def Func(df:pd.DataFrame, option:int):
abcd
我想对所有子数据帧运行这个函数。
from pqdm.processes import pqdm
results = pqdm((DataFrame.groupby(["col1", "col2"]), 3) , Func, n_jobs=16)
#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# this part is the problem. It have to a single generator.
“DataFrame.groupby([“col1”,“col2”])”是生成器。 3 只是一个值。
如何连接生成器和值?
我需要一个新的生成器,它返回 (subgroup1,3), (subgroup2,3),(subgroup3,3),(subgroup4,3),.....
from pqdm.processes import pqdm
from itertools import repeat
results = pqdm(zip(DataFrame.groupby(["col1", "col2"]), repeat(3)) , Func, n_jobs=16)
使用 zip 和 itertools.repeat 作为值是解决方案。