这个问题在这里已经有了答案:查找和替换列表中的元素(10 个答案)3 年前关闭。
我有一个列表,我想用 None 替换值,其中 condition() 返回 True。
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
例如,如果条件检查 bool(item%2) 应该返回:
[None, 1, None, 3, None, 5, None, 7, None, 9, None]
最有效的方法是什么?
in-place
替换比较,请查看此 answer
使用列表推导构建一个新列表:
new_items = [x if x % 2 else None for x in items]
如果需要,您可以就地修改原始列表,但实际上并不能节省时间:
items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for index, item in enumerate(items):
if not (item % 2):
items[index] = None
以下是 (Python 3.6.3) 时序演示非省时:
In [1]: %%timeit
...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
...: for index, item in enumerate(items):
...: if not (item % 2):
...: items[index] = None
...:
1.06 µs ± 33.7 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
In [2]: %%timeit
...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
...: new_items = [x if x % 2 else None for x in items]
...:
891 ns ± 13.6 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
和 Python 2.7.6 时间:
In [1]: %%timeit
...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
...: for index, item in enumerate(items):
...: if not (item % 2):
...: items[index] = None
...:
1000000 loops, best of 3: 1.27 µs per loop
In [2]: %%timeit
...: items = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
...: new_items = [x if x % 2 else None for x in items]
...:
1000000 loops, best of 3: 1.14 µs per loop
ls = [x if (condition) else None for x in ls]
这是另一种方式:
>>> L = range (11)
>>> map(lambda x: x if x%2 else None, L)
[None, 1, None, 3, None, 5, None, 7, None, 9, None]
对OP在评论中提出的一个附带问题进行评论,即:
如果我有一个生成器从范围(11)而不是列表中产生值怎么办。是否可以替换生成器中的值?
当然,这很容易......:
def replaceiniter(it, predicate, replacement=None):
for item in it:
if predicate(item): yield replacement
else: yield item
只需将任何可迭代对象(包括调用生成器的结果)作为第一个参数传递,判断一个值是否必须被替换为第二个参数的谓词,然后让 'er rip。
例如:
>>> list(replaceiniter(xrange(11), lambda x: x%2))
[0, None, 2, None, 4, None, 6, None, 8, None, 10]
>>> L = range (11)
>>> [ x if x%2 == 1 else None for x in L ]
[None, 1, None, 3, None, 5, None, 7, None, 9, None]
如果您想替换值,您可以通过分配给原始列表的整个切片,使用列表理解中的值更新原始列表。
data = [*range(11)] # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
id_before = id(data)
data[:] = [x if x % 2 else None for x in data]
data
# Out: [None, 1, None, 3, None, 5, None, 7, None, 9, None]
id_before == id(data) # check if list is still the same
# Out: True
如果您有多个名称指向原始列表,例如您在更改列表之前编写了 data2=data
,并且您跳过了分配给 data
的切片符号,则 data
将重新绑定以指向新创建的列表,而 {4 } 仍然指向原始未更改的列表。
data = [*range(11)] # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
data2 = data
id_before = id(data)
data = [x if x % 2 else None for x in data] # no [:] here
data
# Out: [None, 1, None, 3, None, 5, None, 7, None, 9, None]
id_before == id(data) # check if list is still the same
# Out: False
data2
# Out: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
注意:这不是通常更喜欢一个而不是另一个的建议(是否更改列表),但是您应该注意的行为。
这可能会有所帮助...
test_list = [5, 8]
test_list[0] = None
print test_list
#prints [None, 8]
不定期副业成功案例分享
enumerate()
会产生少量开销,但如果无法接受,可以手动跟踪索引。 @ak:我不明白这个问题。imap()
不是就地操作。def replace(items, a, b): return [b if x == a else x for x in items]