使用列表理解填充熊貓值字典

Question

有沒有辦法用列表理解替換這個語法？

for w in loc:
    dict_filter_data[w] = df.loc[df['location'] == w]

如果可以的話，會不會更快？

Answer 1

你可以做：

dict_filter_data = dict(df.loc[df['location'].isin(loc)]
                          .groupby('location').__iter__()
                       )

如果loc包含所有唯一的location值，那么您只需要：

dict_filter_data= dict(df.groupby('location').__iter__())

請注意，強烈建議在此處使用 groupby，它比使用 for 循環快得多。 但你可以這樣做：

dict_filter_data = {w : df.loc[df['location'] == w] for w in loc}

如果你想更新dict_filter_data （不是開始為空） ：

dict_filter_data.update(dict(df.loc[df['location'].isin(loc)]
                               .groupby('location').__iter__()
                            )
                       )

或者

dict_filter_data = dict(dict_filter_data, 
                        **dict(df.loc[df['location'].isin(loc)]
                                 .groupby('location').__iter__()
                              )
                        )