使用Python替换列表中的值

时间:2009-10-08 19:54:18

标签: python list performance replace

我有一个列表,我想用None替换值,其中condition()返回True。

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

例如,如果条件检查bool(项目%2)应该返回:

[None, 1, None, 3, None, 5, None, 7, None, 9, None]

最有效的方法是什么?

7 个答案:

答案 0 :(得分:164)

使用列表推导构建新列表:

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

答案 1 :(得分:61)

ls = [x if (condition) else None for x in ls]

答案 2 :(得分:10)

对OP在评论中提出的问题进行评论,即:

  如果我有一台能够产生的发电机怎么办?   范围(11)而不是a的值   名单。是否可以更换   生成器中的值?

当然,这很简单......:

def replaceiniter(it, predicate, replacement=None):
  for item in it:
    if predicate(item): yield replacement
    else: yield item

只需传递任何可迭代(包括调用生成器的结果)作为第一个arg,谓词决定是否必须将值替换为第二个arg,然后让我们翻录。

例如:

>>> list(replaceiniter(xrange(11), lambda x: x%2))
[0, None, 2, None, 4, None, 6, None, 8, None, 10]

答案 3 :(得分:10)

这是另一种方式:

>>> L = range (11)
>>> map(lambda x: x if x%2 else None, L)
[None, 1, None, 3, None, 5, None, 7, None, 9, None]

答案 4 :(得分:2)

>>> 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]

答案 5 :(得分:0)

如果您想要替换值,您可以 使用列表中的值更新原始列表 通过分配整个原始片段来理解。

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 并跳过切片表示法以分配给datadata将重新绑定以指向新创建的列表,而data2仍然指向原始未更改的列表。

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]

注意:这不建议通常更喜欢一个而不是另一个 (更改列表与否),但您应该注意的行为。

答案 6 :(得分:-1)

这可能会有所帮助......

test_list = [5, 8]
test_list[0] = None
print test_list
#prints [None, 8]