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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

例子：