Basic snippet for searching items

search_pandas.py

def search_item(dataframe, name, query, na=False, case=False, regex=True):

    idx = pd.Series([False]*len(dataframe))

    # For each item in the query look for the item and collect the documents ids it pertains to
    for q in query:
        matches = dataframe[text_column].str.contains(q, na=False, case=False, regex=True)
        idx = (idx) | (matches)

    dataframe.loc[idx, name] = 1
    dataframe.loc[:, name] = dataframe.loc[:, name].fillna(0)
    return dataframe


def make_dataframe():
    df = pd.DataFrame(dict(a=['a', 'b', 'c'], b=['A', 'B', 'C'])
    return df