Python分解超长IF语句

Question

如何简化或分解Python pandas中的超长IF条件？

my_dataframes = {'d1': d1, 'd2': d2,'d3': d3}
good_dataframes = []
for df_name, df in my_dataframes.items():
    if ((df.loc[1:4, 'test'] <= 6).all() 
        and (df.loc[5:9, 'dif'] < 9).all()) or ((df.loc[1:5, 'test'] <= 6).all() 
                                                      and (df.loc[5:8, 'dif'] < 8).all()) or ((df.loc[1:8, 'test'] <= 6).all() 
                                                                                                    and (df.loc[5:8, 'dif'] < 9).all()):
        good_dataframes.append(df_name)

对我来说，挑战是正确的缩进

Answer 1

定义一个包含条件的函数。

def is_good_df(df):
    return (((df.loc[1:4, 'test'] <= 6).all()
             and (df.loc[5:9, 'dif'] < 9).all())
            or ((df.loc[1:5, 'test'] <= 6).all()
                and (df.loc[5:8, 'dif'] < 8).all())
            or ((df.loc[1:8, 'test'] <= 6).all()
                and (df.loc[5:8, 'dif'] < 9).all()))

一个好的 IDE 应该有助于获得正确的缩进。

请注意，在某些 .all 调用之后，您还缺少 ()。

然后你就可以了

my_dataframes = {'d1': d1, 'd2': d2,'d3': d3}
good_dataframes = [df_name for df, df_name in my_dataframes.items() if is_good_df(df)]