在dask中搜索行后获取列值

Question

我有一个pandas数据帧，我使用dask的from_pandas函数将其转换为dask数据帧。它有3列，即col1、col2和col3

现在，我正在使用daskdf[(daskdf.col1 == v1) & (daskdf.col2 == v2)]搜索特定行，其中v1和v2是我正在搜索的值。但是，当我尝试使用daskdf[(daskdf.col1 == v1) & (daskdf.col2 == v2)]['col3']获取col3的值时，它给了我一个dask系列结构，而不是列值

在熊猫中，我可以做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()