python - 在dask中搜索行后获取列值

Question

我有一个 Pandas 数据帧，我使用 from_pandas 将其转换为 dask 数据帧dask 的功能。它有 3 列，即 col1 , col2和 col3 .
现在我正在使用 daskdf[(daskdf.col1 == v1) & (daskdf.col2 == v2)] 搜索特定行哪里v1和 v2是我正在寻找的值。但是当我尝试获取 col3 的值时使用 daskdf[(daskdf.col1 == v1) & (daskdf.col2 == v2)]['col3']它给了我一个 dask 系列结构而不是列值。
在 Pandas 我可以做 pandasdf[(pandasdf.col1 == v1) & (pandasdf.col2 == v2)]['col3'].tolist() .如何获取 col3 的值这里 ？

Answer 1

运行 .compute将转换 dask对象进入 pandas对象，因此以下将给出所需的结果:

result = daskdf[(daskdf.col1 == v1) & (daskdf.col2 == v2)]['col3'].compute().tolist()