Группировать столбец df на основе двух других столбцов ⇐ Python

[Anonymous]

У меня есть файл df, в котором есть три столбца: «Регион», «Страна» и «AREA_CODE».

Код: Выделить всё

Region      Country     AREA_CODE
===================================
AMER       US             A1
AMER       CANADA         A1
AMER       US             B1
AMER       US             A1

Я хочу получить выходные данные в виде списка AREA_CODE для каждой страны в каждом регионе со значением списка «ВСЕ». что-то вроде

Код: Выделить всё

{
"AMER": {
"US": ["ALL", "A1", "B1"],
"CANADA": ["ALL", "A1"]
}
}

До сих пор я пытался группировать по столбцу региона и страны, а затем пытался сгруппировать и объединить их по AREA_CODE, это выдавало ошибку

Код: Выделить всё

df.drop_duplicates().groupby(["Region", "Country"]).groupby("Country")['AREA_CODE'].agg(lambda x: ["ALL"]+sorted(x.unique().tolist())).to_dict()

Может ли кто-нибудь помочь мне с этим?
Спасибо,

Подробнее здесь: https://stackoverflow.com/questions/791 ... er-columns