def get_data_for_subid(raw_data, sub_ids):
    """Filters data for a particular subscription id or a list of subscription ids
    Args:
        raw_data: raw usage dataframe
        sub_ids: subscription id or a list of subscription ids
    """

    if isinstance(sub_ids, (list,)):
        return raw_data[raw_data.SubscriptionGuid.isin(sub_ids)]
    else:
        return raw_data[raw_data.SubscriptionGuid == sub_ids]