Как сгруппировать районы по странам в цикле for в python?

Question

У меня есть датафрейм со странами и районами, я хотел бы сгруппировать все районы по странам.

Это мой код:

sa_d = []
ma_d = []
za_d = []
mo_d = []
for d, f in dn['interview_country', 'interview_district']:
    if d == 'South Africa':
      sa_d.append(f)
    elif d == 'Malawi':
      ma_d.append(f)
    elif d == 'Zambia':
      za_d.append(f)
    elif d == 'Mozambique':
      zo_d.append(f)

Я хотел бы получить список районов для каждой страны в качестве вывода. Вместо этого я получаю сообщение об ошибке:

KeyError: ('interview_country', 'interview_district')

Answer 1

Добавьте еще одну пару скобок (при условии, что для данных pandas):

sa_d = []
ma_d = []
za_d = []
mo_d = []
for d, f in dn[['interview_country', 'interview_district']].values.tolist():
    if d == 'South Africa':
      sa_d.append(f)
    elif d == 'Malawi':
      ma_d.append(f)
    elif d == 'Zambia':
      za_d.append(f)
    elif d == 'Mozambique':
      zo_d.append(f)