Получить номера из файла, соответствующие заданным регионам, указанным в файле - PullRequest
2 голосов
/ 27 мая 2019

Ниже мой фрейм данных:

 Sno       Name    Region         Num
0    1      Rubin    Indore  79744001550
1    2      Rahul     Delhi  89824304549
2    3      Rohit     Noida  91611611478
3    4     Chirag     Delhi  85879761557
4    5       Shan    Bharat  95604535786
5    6      Jordi    Russia  80777784005
6    7         El    Russia  70008700104
7    8       Nino     Spain  87707101233
8    9       Mark       USA  98271377772
9   10  Pattinson  Hawk Eye  87888888889

Получить номера и сохранить их по регионам из данного файла CSV.

delhi_list = []

for i in range(len(data)):
   if data.loc[i]['Region'] == 'Delhi':
     delhi_list.append(data.loc[i]['Num'])

delhi_list = []

for i in range(len(data)):
   if data.loc[i]['Region'] == 'Delhi':
     delhi_list.append(data.loc[i]['Num'])

Я получаю результаты, но хочу получить данные, используя словарь в python. Могу ли я использовать это?

1 Ответ

2 голосов
/ 28 мая 2019

IIUC, вы можете использовать groupby, применить агрегацию list, затем использовать to_dict:

data.groupby('Region')['Num'].apply(list).to_dict()

[out]

{'Bharat': [95604535786],
 'Delhi': [89824304549, 85879761557],
 'Hawk Eye': [87888888889],
 'Indore': [79744001550],
 'Noida': [91611611478],
 'Russia': [80777784005, 70008700104],
 'Spain': [87707101233],
 'USA': [98271377772]}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...