将生成器和值结合起来[关闭]

问题描述 投票:0回答:1

我必须将带有值的生成器作为参数传递给特定函数。

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),.....

python generator
1个回答
0
投票
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 作为值是解决方案。

© www.soinside.com 2019 - 2024. All rights reserved.