在整理航班数据时,需要整理成key-value方便处理。比如,北京有大兴机场、首都机场。希望处理成'北京':['大兴机场,'首都机场'']的形式。
airports = ['北京/首都国际机场','北京/首都机场','北京/大兴国际机场','北京/大兴机场','天津/滨海国际机场','天津/滨海机场','石家庄/正定国际机场','石家庄/正定机场']
使用python中字典的.setdefault方法,将字典中的值设置为列表的形式。具体代码如下。
airports_stations_dict = {}
for item in airports:
index = item.split('/')[0]
airports_stations_dict.setdefault(index, []).append(item)