如何将含条件判断的多行Python代码转换为单行代码？

原多行代码： #

for k, v in FLOW_GROUPS.items():
    if k == '':
        [pd.Series(df_by_fulfillment_flow[v].sum(axis=1), name=k)]
    else:
        [pd.Series(df_by_fulfillment_flow[v].mean(axis=1), name=k)]

现有基础列表推导式： #

[pd.Series(df_by_fulfillment_flow[v].sum(axis=1), name=k) for k, v in FLOW_GROUPS.items()]

添加条件判断后的单行代码： #

只需在列表推导式的元素生成部分嵌入三元表达式，根据k的值选择对应的聚合方法即可：

[pd.Series(df_by_fulfillment_flow[v].sum(axis=1) if k == '' else df_by_fulfillment_flow[v].mean(axis=1), name=k) for k, v in FLOW_GROUPS.items()]

逻辑说明： #

• 三元表达式df_by_fulfillment_flow[v].sum(axis=1) if k == '' else df_by_fulfillment_flow[v].mean(axis=1)替代了原代码的分支判断，实现了根据k是否为空字符串选择求和或求均值的逻辑。
• 整体保留列表推导式的结构，遍历FLOW_GROUPS.items()的同时完成条件判断与pd.Series对象的生成。

内容的提问来源于stack exchange，提问作者Vibhor Gupta