内联循环

时间:2014-12-10 21:58:45

标签: python syntax list-comprehension

我正在尝试学习整齐的pythonic做事方式,并且想知道为什么我的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:循环根据vm中的q下一个数字来驱逐{{1}}中的数字。

4 个答案:

答案 0 :(得分:30)

您使用的是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方法总是返回。

答案 1 :(得分:2)

你可以使用enumerate保持元素的ind / index是vm,如果你使vm成为set,你也会0(1)次查找:

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

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

答案 2 :(得分:2)

你的列表comphresnion会工作,但会返回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]

答案 3 :(得分:1)

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

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

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

输出:

[99999, 99999, 99999, 99999, 0, 1, 2, 0]
[99999, 99999, 99999, 99999, 0, 1, 2, 0]

您可以将p引用为直接输出,而不是在列表推导中使用append(),并在LC中使用q.index(v)99999

不确定这是否是故意的,但请注意q.index(v)只会发现v的第一次出现,即使您在q中有多个也是如此。如果您想获取v中所有q的索引,请考虑使用enumerator和已访问过的indexes

列表

这些行中的东西(伪代码):

visited = []
for i, v in enumerator(vm):
   if i not in visited:
       p.append(q.index(v))
   else:
       p.append(q.index(v,max(visited))) # this line should only check for v in q after the index of max(visited)
   visited.append(i)