class HistogramTable2DictTableConverter:

    @staticmethod
    def convertHistogramTable2DictTable(symptomHistogramByBatchcodeTable):
        vax_lot_columns = symptomHistogramByBatchcodeTable.index.names.difference(['SYMPTOM'])
        return (symptomHistogramByBatchcodeTable
                    .groupby(vax_lot_columns + ['COUNTRY'])
                    .agg(lambda histogram_with_vax_lots: HistogramTable2DictTableConverter._histogram_to_json(histogram_with_vax_lots, vax_lot_columns))
                    .reset_index(level = 'COUNTRY')
                    [['SYMPTOM_COUNT_BY_VAX_LOT', 'COUNTRY']])

    @staticmethod
    def _histogram_to_json(histogram_with_vax_lots, vax_lot_columns):
        histogram = histogram_with_vax_lots.reset_index(level = vax_lot_columns, drop = True)
        return histogram.to_dict()