def indexes_for_key(self, key, devptr=0): """ Return all index values for given key. Parameters ---------- key : str key whose values should be returned devptr : GPU memory pointer Where index values will be written. Must be able to hold int32 values for this key. Examples -------- .. code-block:: python import nvcategory c = nvcategory.to_device(["eee","aaa","eee","dddd"]) print(c.indexes_for_key('aaa')) print(c.indexes_for_key('eee')) Output: .. code-block:: python [1] [0, 2] """ return pyniNVCategory.n_get_indexes_for_key(self.m_cptr, key, devptr)
def indexes_for_key(self, key, devptr=0): """ Return all index values for given key. Parameters ---------- key : str or number key whose values should be returned devptr : GPU memory pointer or ndarray Where index values will be written. Must be able to hold int32 values for this key. Examples -------- >>> import nvcategory >>> c = nvcategory.to_device(["eee","aaa","eee","dddd"]) >>> print(c.indexes_for_key('aaa')) [1] >>> print(c.indexes_for_key('eee')) [0, 2] >>> import numpy as np >>> narr = np.array([2, 1, 1.25, 1.5, 1, 1.25, 1, 1, 2]) >>> nc = nvcategory.from_numbers(narr) >>> count = nc.indexes_for_key(1) >>> idxs = np.empty([count], dtype=np.int32) >>> count = nc.indexes_for_key(1, idxs) >>> idxs.tolist() [1, 4, 6, 7] """ return pyniNVCategory.n_get_indexes_for_key(self.m_cptr, key, devptr)