Python中列表理解中的多个If/else

Question

我有这个

s = ['son','abc','pro','bro']
b = ['son','bro']
c = ['pro','quo']

预期的输出是这样的。输出中的项目是 index(item_in_s)，如果它存在于列表 b 中。或者 index(item_in_s)+10 如果项目在 c 中。

[0,12,3]

我试过这个：

index_list = [s.index(item) if item in b else s.index(item)+10 if item in c for item in s]
print(index)

但显然这是一个语法错误。所以我尝试了这个：

index_list = [s.index(item) if item in b else s.index(item)+10 for item in s if item in c]
    print(index)

输出：

[12]

这只是改变了整个逻辑。虽然我可以这样做

fin = [s.index(item) if item in b else s.index(item)+10 if item in c  else '' for item in s]
fin = [item for item in fin if item!='']
print(fin)

获得的期望输出：

[0, 12, 3]

但是我如何在列表推导本身中获得我想要的东西，或者在列表推导中是否有 else continue 之类的东西？

Answer 1

index_list = [s.index(item) if item in b else s.index(item) + 10 for item in s if item in b or item in c]

我们确保它在 b 或 c 中，然后索引将是其中一种情况。

Answer 2

已经给出了很好的答案，这里还没有提到：

fin = [item for item in ([s.index(item) if item in b else s.index(item)+10 if item in c  else '' for item in s]) if item!='']
print(fin)

基本上它是原始2行代码的组合：

fin = [s.index(item) if item in b else s.index(item)+10 if item in c  else '' for item in s]
fin = [item for item in fin if item!='']

不一定是“更快或更好”，只是以前没有给出的组合。 在进行列表推导时，您总是面临迭代次数超出实际需要的风险。

更好的解决方案是for 循环，在另一个答案中给出。

Answer 3

您的解决方案故障可以通过避免列表 b&c 中不存在的元素来解决。

您可以通过创建新列表并应用简单的集合操作来做到这一点

检查您的解决方案中的这个小改动。

fin = [s.index(item) if item in b else s.index(item)+10 if item in c  else '' for item in list(set(b+c)&set(s))]

通过这样做，您的条件语句 else 将永远不会执行您正在迭代的原因列表仅具有 b 或 clist 元素/strong> 仅限。

Answer 4

从根本上说，列表理解会迫使您非常低效：

>>> [i if item in b else i + 10 if item in c else None for i, item in enumerate(s) if item in b or item in c]
[0, 12, 3]

这必须检查b 和c 中的成员item 在最坏的情况下每个 如果你想要那个输出。相反，只需使用 for 循环：

>>> index_list = []
>>> for i, item in enumerate(s):
...     if item in b:
...         index_list.append(i)
...     elif item in c:
...         index_list.append(i + 10)
...
>>> index_list
[0, 12, 3]
>>>

简单、易读、直接且 Pythonic。