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)