我正在尝试学习整齐的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}}中的数字。
答案 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)