当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python cudf.MultiIndex.get_loc用法及代码示例


用法:

MultiIndex.get_loc(key, method=None, tolerance=None)

获取标签或标签元组的位置。

该位置以整数/切片或布尔掩码的形式返回。

参数

key标签或标签元组(每个级别一个)
methodNone

返回

locint,切片对象或布尔掩码
  • 如果索引是唯一的,则搜索结果是唯一的,返回一个 int.
  • 如果 index 是单调的,则 index 作为切片对象返回。
  • 否则, cudf 会尽最大努力将搜索结果转换为切片对象,如果不这样做,将返回一个布尔掩码。请注意,在某些情况下,这可能会偏离 Pandas 的行为。

例子

>>> import cudf
>>> mi = cudf.MultiIndex.from_tuples(
...     [('a', 'd'), ('b', 'e'), ('b', 'f')])
>>> mi.get_loc('b')
slice(1, 3, None)
>>> mi.get_loc(('b', 'e'))
1
>>> non_monotonic_non_unique_idx = cudf.MultiIndex.from_tuples(
...     [('c', 'd'), ('b', 'e'), ('a', 'f'), ('b', 'e')])
>>> non_monotonic_non_unique_idx.get_loc('b') # differ from pandas
slice(1, 4, 2)

Pandas 兼容性说明

多索引。get_loc

该函数的返回类型可能与 Pandas 提供的方法不同。如果索引既不是按字典顺序排序也不是唯一的,则尽最大努力将找到的索引强制转换为切片。例如:

>>> import pandas as pd
>>> import cudf
>>> x = pd.MultiIndex.from_tuples([
...     (2, 1, 1), (1, 2, 3), (1, 2, 1),
...     (1, 1, 1), (1, 1, 1), (2, 2, 1),
... ])
>>> x.get_loc(1)
array([False,  True,  True,  True,  True, False])
>>> cudf.from_pandas(x).get_loc(1)
slice(1, 5, 1)

相关用法


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