def extract_cols(data, extract_str):
    """
    he function returns all columns with the given string
    in the column name.
    """
    return data[[col for col in data.columns if extract_str in col]]