A way to check if a value exists in a single row in a numpy array

numpy_row_contains.py

def row_contains(data, query):
    """
    Checks row of the matrix for the given condition
    and returns a list of rows that satisfies the condition
    """
    condition = np.any(query, axis=1)
    return np.compress(condition, data, axis=0)