python - 从文件中检索与文件中指定的给定区域相对应的数字

Question

下面是我的数据框:

 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 中使用字典来实现数据。我可以使用它吗？

Answer 1

IIUC，你可以使用groupby ，应用 list 聚合然后使用 to_dict :

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

[输出]

{'Bharat': [95604535786],
 'Delhi': [89824304549, 85879761557],
 'Hawk Eye': [87888888889],
 'Indore': [79744001550],
 'Noida': [91611611478],
 'Russia': [80777784005, 70008700104],
 'Spain': [87707101233],
 'USA': [98271377772]}