Python cudf.DataFrame.query用法及代码示例

用法: DataFrame.query(expr, local_dict=None)

使用 Numba 使用布尔表达式查询以编译 GPU 内核。

请参阅 pandas.DataFrame.query。

参数：

expr：str

一个布尔表达式。表达式中的名称指的是列。 index 可用于代替索引名称，但 MultiIndex 不支持此函数。

以 @ 开头的名称指的是 Python 变量。

如果任何输入值是null，则无论表达式如何，输出值都将为null。

local_dict：dict

包含要在查询中使用的局部变量。

filtered：DataFrame

例子：

>>> df = cudf.DataFrame({
...     "a": [1, 2, 2],
...     "b": [3, 4, 5],
... })
>>> expr = "(a == 2 and b == 4) or (b == 3)"
>>> df.query(expr)
   a  b
0  1  3
1  2  4

日期时间条件：

>>> import numpy as np
>>> import datetime
>>> df = cudf.DataFrame()
>>> data = np.array(['2018-10-07', '2018-10-08'], dtype='datetime64')
>>> df['datetimes'] = data
>>> search_date = datetime.datetime.strptime('2018-10-08', '%Y-%m-%d')
>>> df.query('datetimes==@search_date')
   datetimes
1 2018-10-08

使用local_dict：

>>> import numpy as np
>>> import datetime
>>> df = cudf.DataFrame()
>>> data = np.array(['2018-10-07', '2018-10-08'], dtype='datetime64')
>>> df['datetimes'] = data
>>> search_date2 = datetime.datetime.strptime('2018-10-08', '%Y-%m-%d')
>>> df.query('datetimes==@search_date',
...          local_dict={'search_date': search_date2})
   datetimes
1 2018-10-08

相关用法

注：本文由纯净天空筛选整理自rapids.ai大神的英文原创作品 cudf.DataFrame.query。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

例子：