pyyeti.locate.find_rows

pyyeti.locate.find_rows(matrix, row)[source]

Get True/False vector indicating where row occurs in matrix

Parameters:

  • matrix (array) – 2d numpy array.

  • row (array) – 1d numpy array.

Returns:

1d ndarray – True/False vector with True indicating where row occurs in matrix. Will be all False if row does not occur in matrix or if len(row) != cols(matrix).

Examples

>>> import numpy as np
>>> from pyyeti import locate
>>> mat = np.array([[7, 3], [6, 8], [4, 0],
...                 [9, 2], [1, 5], [6, 8]])
>>> locate.find_rows(mat,np.array([1, 2]))
array([False, False, False, False, False, False], dtype=bool)
>>> pv = locate.find_rows(mat,np.array([6, 8]))
>>> pv
array([False,  True, False, False, False,  True], dtype=bool)
>>> mat[pv, :]
array([[6, 8],
       [6, 8]])