使用嵌套列表理解提高执行速度

Question

下面是我花了很长时间执行的代码。如何在Python的列表理解中实现它以提高速度和效率

buildings=[]
for bi in range(1449):
    for si in range (16):
        for m in range(3):
            a= train[(train['building_id']==bi)&(train['site_id']==si)&(train['meter']==m)]
            if not a.empty:

                buildings.append(a.values)

Answer 1

如果没有您的示例数据，很难判断这是否正确，但这在理论上应该是可行的：

buildings = [
             x for x in
                 [
                 train[(train['building_id']==bi)&(train['site_id']==si)&(train['meter']==m)].values
                 for bi in range(1449)
                 for si in range(16)
                 for m in range(3)
                 ]
            if not x.empty
            ]