您的代码有几个问题。
首先,分配None
给列表元素不会删除它:
>>> l=[1,2,3,4,5,6,6,7,8,9]
>>> len(l)
10
>>> l[l.index(5)]=None
>>> l
[1, 2, 3, 4, None, 6, 6, 7, 8, 9]
>>> len(l)
10
其次,使用索引来查找要更改的元素根本不是有效的方法。
您可以使用枚举,但您仍然需要循环删除这些None
值。
for i,a in enumerate(b):
# Do many things with a
b[i]=C(modified_a)
b[i]=None
c=[e for e in b if e is not None]
您可以使用列表推导将新的“a”值复制到 c 列表中,然后删除 b:
c=[do_many_things(a) for a in b]
del b # will still occupy memory if not deleted...
或者,如果您希望 b 被原地修改,您可以使用slice assignment:
b[:]=[do_many_things(a) for a in b]
切片分配以这种方式工作:
#shorted a list
>>> b=[1,2,3,4,5,6,7,8,9]
>>> b[2:7]=[None]
>>> b
[1, 2, None, 8, 9]
#expand a list
>>> c=[1,2,3]
>>> c[1:1]=[22,33,44]
>>> c
[1, 22, 33, 44, 2, 3]
# modify in place
>>> c=[1,2,3,4,5,6,7]
>>> c[0:7]=[11,12,13,14,15,16,17]
>>> c
[11, 12, 13, 14, 15, 16, 17]
您可以在列表理解中使用它,如下所示:
>>> c=list(range(int(1e6)))
>>> c[:]=[e for e in c if e<10]
>>> c
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
其中一条评论指出,切片分配并没有完全修改到位;生成一个临时列表。那是真实的。但是,让我们看看这里的总时间:
import time
import random
fmt='\t{:25}{:.5f} seconds'
count=int(1e5)
a=[random.random() for i in range(count)]
b=[e for e in a]
t1=time.time()
for e in b:
if e<0.5: b[b.index(e)]=None
c=[e for e in b if e is not None]
print(fmt.format('index, None',time.time()-t1))
b=[e for e in a]
t1=time.time()
for e in b[:]:
if e<0.5: del b[b.index(e)]
print(fmt.format('index, del',time.time()-t1))
b=[e for e in a]
t1=time.time()
for i,e in enumerate(b[:]):
if e<0.5: b[i]=None
c=[e for e in b if e is not None]
print(fmt.format('enumerate, copy',time.time()-t1))
t1=time.time()
c=[e for e in a if e<.5]
del a
print(fmt.format('c=',time.time()-t1))
b=[e for e in a]
t1=time.time()
b[:]=[e for e in b if e<0.5]
print(fmt.format('a[:]=',time.time()-t1))
在我的电脑上,打印这个:
index, None 87.30604 seconds
index, del 28.02836 seconds
enumerate, copy 0.02923 seconds
c= 0.00862 seconds
a[:]= 0.00824 seconds
或者,如果这没有帮助,请使用 numpy 以获得更优化的数组选项。