isin#

bamengine.ops.isin(element, test_elements)[source]#

Test whether each element is in test_elements.

Parameters:

element (array) – Input array.
test_elements (array) – Values to test against.

Returns:

True where element is in test_elements.

Return type:

bool array

Examples

>>> import bamengine.ops as ops
>>> firm_ids = np.array([0, 1, 2, 3, 4])
>>> bankrupt_ids = np.array([1, 3])
>>> is_bankrupt = ops.isin(firm_ids, bankrupt_ids)
>>> # is_bankrupt: [False, True, False, True, False]