如何将条件循环转换为列表理解？

问题描述

我有一个带有 for 条件的 if 循环，我想将其转换为列表理解，循环看起来像这样

if options.accounts.lower() == "all":
    orgPaginator = org.get_paginator('list_accounts_for_parent')
    for orgPages in orgPaginator.paginate(ParentId=src_ou_id):
        for account in orgPages['Accounts']:
            acc.append(account["Id"])
else:
    for account in options.accounts.split(','):
        acc.append(account)

这就是我现在所拥有的我可以弄清楚其他部分，但我在转换它的第一部分时遇到了问题

[<??> if options.accounts.lower() == "all" else acc.append(account) for account in options.accounts.split(',')]

有人可以就如何转换第一部分提供一些指导吗？

解决方法

if options.accounts.lower() == "all":
    acc += [account["Id"] for account in [orgPages["Accounts"] for orgPages in org.get_paginator('list_accounts_for_parent').paginate(ParentId=src_ou_id)]]
else:
    acc += [account for account in options.accounts.split(',')]

这应该可以。如果可能的话，我不认为纯粹的列表理解方法会更好。

您的 <??> 似乎是您真正想要放入列表的内容：

[f(x) for x in iterable]

在这种情况下，f(x) 将是 x[...]。

,

我/我们可以做到这一点（编辑：在@ShadowRanger 的帮助下）：

acc += [p['Accounts'] for p in org.get_paginator('list_accounts_for_parent').paginate(ParentId=src_ou_id)] \
        if options.accounts.lower() == "all" else \
        options.accounts.split(',')