def GP(data, init_haps, positions, N, r, times, numThreads=1, pool=None):
if numThreads == 1:
results = [
GP.singleLocus(data[:, (i,), :], init_haps[:, (i,)], N, r, times, pos)
for i, pos in enumerate(positions)
]
else:
params = [
{"X": data[:, (i,), :], "initHaps": init_haps[:, (i,)], "N": N, "r": r, "times": times, "pos": pos}
for i, pos in enumerate(positions)
]
results = pool.map(gpHelper, params)
df = pd.DataFrame(map(lambda x: x.values(), results), columns=results[0].keys()).astype(float)
df.index = df.pos.astype(int)
df.drop("pos", axis=1, inplace=True)
return df
def gpHelper(args):
return GP.singleLocus(**args)