Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def closure(df, func, *args, **kwargs):
axis = kwargs.get("axis", 0)
if axis == 'index':
axis = 0
elif axis == 'columns':
axis = 1
opposite_axis = 1 - axis
chunks = _chunk(df.shape[opposite_axis], nb_workers)
object_id = plasma_client.put(df)
with _ProcessPoolExecutor(max_workers=nb_workers) as executor:
futures = [
executor.submit(_DataFrame.worker_apply,
plasma_store_name, object_id,
chunk, func, progress_bar,
*args, **kwargs)
for index, chunk in enumerate(chunks)
]
result = _pd.concat([
plasma_client.get(future.result())
for future in futures
], copy=False)
return result
return closure
def closure(df, func):
chunks = _chunk(df.shape[0], nb_workers)
object_id = plasma_client.put(df)
with _ProcessPoolExecutor(max_workers=nb_workers) as executor:
futures = [
executor.submit(_DataFrame.worker_applymap,
plasma_store_name, object_id,
chunk, func, progress_bar)
for index, chunk in enumerate(chunks)
]
result = _pd.concat([
plasma_client.get(future.result())
for future in futures
], copy=False)
return result
return closure