使用Python替换列表中的值

Question

我有一个列表，我想用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]

最有效的方法是什么？

Answer 1

创建一个包含列表理解的新列表：

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)

以及Python2.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

Answer 2

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

Answer 3

另一种方法是：

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