内联for循环

2024-04-20 09:01:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试学习整洁的python做事方式,并想知道为什么我的for循环不能这样重构:

q  = [1, 2, 3, 4, 1, 2, 5, 1, 2, 3, 4, 5]
vm = [-1, -1, -1, -1]

for v in vm:
    if v in q:
        p.append(q.index(v))
    else:
        p.append(99999)

vm[p.index(max(p))] = i

我尝试将for循环替换为:

[p.append(q.index(v)) if v in q else p.append(99999) for v in vm]

但没用。for v in vm:循环根据数字在q中的下一个出现时间从vm中逐出数字。


Tags: inforindexif方式时间vm数字
3条回答

您使用的是Python中的list comprehension,而不是内联for循环(尽管它与一个类似)。你可以把你的循环写成一个列表理解,就像这样:

p = [q.index(v) if v in q else 99999 for v in vm]

使用列表理解时,不要调用list.append,因为列表是根据理解本身构造的。列表中的每个项都是由for关键字左侧的表达式返回的,在本例中是q.index(v) if v in q else 99999。顺便说一下,如果在理解中使用list.append,那么您将得到一个None值的列表,因为append方法总是返回这个值。

您可以使用enumerate保持元素的ind/索引在vm中,如果您进行vmaset的话,您还将有0(1)查找:

vm = {-1, -1, -1, -1}

print([ind if q in vm else 9999 for ind,ele in enumerate(vm) ])

列表comphresnion将工作,但将返回list of None,因为append return None:

演示:

>>> a=[]
>>> [ a.append(x) for x in range(10) ]
[None, None, None, None, None, None, None, None, None, None]
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

更好的使用方法如下:

>>> a= [ x for x in range(10) ]
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

相关问题 更多 >