def get_simple_partition(dataframe, case_id_key, no_partitions):
    """
    Get a simple partitioning of a CSV

    Parameters
    --------------
    dataframe
        Dataframe
    case_id_key
        Column that is the case ID
    no_partitions
        Number of partitions

    Returns
    --------------
    dataframe
        Dataframe with a partitioning (@@partition) column
    """
    dataf = dataframe.copy()
    dataf['@@partition'] = dataf[case_id_key].rank(method='dense',
        ascending=False).astype(int) % no_partitions
    return dataf
