如何重构代码以返回字典集合？

Question

def read_data(service_client):
    data = list_data(domain, realm) # This returns a data frame 
    building_data = []
    building_names = {}
    all_buildings = {}

    for elem in data.iterrows():
        building = elem[1]['building_name']
        region_id = elem[1]['region_id']
        bandwith = elem[1]['bandwith']
        building_id = elem[1]['building_id']

       return {
       'Building': building,
       'Region Id': region_id,
       'Bandwith': bandwith,
       'Building Id': building_id,
       }

基本上，在这个例子中，我能够在迭代时返回一个字典值。我也试过打印它和其他的。

我正在设法在每次迭代中存储多个字典值并返回它，而不是只返回一个。有谁知道实现这个目标的方法吗？

Answer 1

您可以用以下内容替换for-循环，以获得list中的所有字典。

naming = {
    'building_name': 'Building',
    'region_id': 'Region Id',
    'bandwith': 'Bandwith',
    'building_id': 'Building Id',
}


return [ 
    row[list(naming.values())].to_dict() 
        for idx, row in data.rename(naming, axis=1).iterrows() 
]