Dask：从已提交的作业中访问已发布的数据集

Question

# Init
import time
import pandas as pd
import numpy as np
from dask.distributed import Client
client = Client()

# Publish data
dataset_name = 'my_dataset'
df_my_dataset = pd.DataFrame(np.ones((2,3)), dtype=np.float32)
client.publish_dataset(df_my_dataset, name=dataset_name)

它在这里：

In [13]: client.list_datasets()                                                                                                           
Out[13]: ('my_dataset',)

为dask创建提交功能。在这里，我想按名称访问已发布的数据集：

# submit function
def get_gate1_rows(df_from_submit):
    return df_from_submit.mean()
    # return df.mean() + my_dataset.mean() #### <<<<<<< How to do this?

最后提交：

# Submit code
df_zeros = np.zeros((2,3), dtype=np.float32)
future = client.submit(get_gate1_rows, df_zeros)
time.sleep(2)
result = future.result()

这会产生-但应为0.5：

In [41]: result                                                                                                                           
Out[41]: 0.0

那么如何从内部快捷任务访问published dataset？

Answer 1

要访问任务中已发布的数据集，您需要get_client：

get_client

（答案是三个1，因为df_zeros.mean（）-> 0，df_my_dataset.mean（）-> 1,1,1）