多进程池使用starmap方式无法按预期工作

2024-09-24 02:21:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图将多个参数传递给一个函数

我已经阅读并尝试了这个解决方案:Python multiprocessing behavior of Pool / starmap 这个:Python multiprocessing pool.map for multiple arguments

但两者似乎都不起作用。在

这是我的代码:

args = [
(1.42,'c',297.39,296,0.0192,0.019,d1,d0),(1.42,'c',297.39,290,0.0192,0.019,d1,d0)
]

with multiprocessing.Pool(processes=2) as pool:
    result = pool.starmap(greeks,args)

它似乎适用于第一组参数,但随后出现以下错误:

^{pr2}$

我甚至不知道为什么会有线程3,因为我只传递了2组参数。 我希望这最终能为大约10-15组arg工作

编辑:完整代码:

import py_vollib.black_scholes_merton.implied_volatility as BSM
from datetime import date
import quantsbin.derivativepricing as qbdp

def greeks(target_value, flag, S, K, r, div,t1,t0):
    if flag == 'c':
        call_put = 'Call'
    else:
        call_put = "Put"

    delta = t1-t0
    delta_annual = delta.days / 365
    d0s = t0.strftime('%Y%m%d')
    d1s = t1.strftime('%Y%m%d')

    sigma = BSM.implied_volatility(target_value, S, K, delta_annual, r, div, flag)

    risk_parameters = {'delta_spot': 0.02, 'delta_vol': 0.02, 'delta_rf_rate': 0.02,
                       'delta_time': 1}  # TODO Fix with next quantsbin release


    equity_o = qbdp.EqOption(option_type=call_put, strike=K, expiry_date=d1s, expiry_type='American')
    engine = equity_o.engine(model="Binomial", spot0=S, pricing_date=d0s,
                             rf_rate=r, yield_div=div, volatility=sigma)

    greeks = engine.risk_parameters(**risk_parameters)
    greeks['IV'] = sigma
    return greeks


d0 = date.today()
d1 = date(2019, 7, 5)


cpu=multiprocessing.cpu_count()-1

t1 = time.time()

args = [(1.42,'c',297.39,296,0.0192,0.019,d1,d0),(1.42,'c',297.39,290,0.0192,0.019,d1,d0)]

with multiprocessing.Pool(processes=cpu) as pool:
     result = pool.starmap(greeks,args)

t2= time.time()

print(result,'{:0.4}'.format(t2-t1))

编辑2:

print(greeks(1.42,'c',297.39,296,0.0192,0.019,d1,d0))

这会起作用并返回:

{'delta': 0.6168499226002772, 'gamma': 0.12714118324230908, 'theta': -0.025340261170400336, 'vega': 2.8379233244261832, 'rho': 1.3659174375347853, 'IV': 0.04311625476862159}

def g(a,b,c):
    return [a+b+c,a-b-c,a+b-c]

a = [(1,5,7),(4,7,2),(7,7,7)]

with multiprocessing.Pool(processes=cpu) as pool:
    result = pool.starmap(g,a)

也如预期的那样工作。在


Tags: datetimeaswithargsresultmultiprocessingd1