python - 对原始 python 列表中的元素子集进行就地修改

Question

我看到列表理解如何擅长返回列表的子集，甚至在没有设置条件的情况下进行全面重新映射。但是，过滤条件将返回一个列表中的值，该列表可能小于过滤后的列表。如何获取返回值的位置并有条件地更改这些元素？

这是一个简单的例子，

>>> v = ['a' , 'b' , 'c' , 'd' , 'f' , 'k' , 'g' , 'a', 'd']
>>> t = ['a', 'k']

我如何修改 t 中 v 的元素以映射到新的新值，例如 'z'，从而产生一些东西喜欢

>>> v = ['z' , 'b' , 'c' , 'd' , 'f' , 'z' , 'g' , 'z', 'd']

有很多关于 SO 的问题，来自 Matlab 的人希望做逻辑索引的等价物，但主要只是为了选择元素而不是修改它们如我所愿。为了修改过滤后的元素，我想做相当于逻辑索引的操作。

我也刚刚发现我不能为 Python 列表提供多个索引。我正在使用 Python 3。

>>> v[ [1,3] ] = 'z'
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: list indices must be integers, not list
>>> v[ 1:3 ] = 'z'
>>> v
['a', 'z', 'd', 'f', 'k', 'g', 'a', 'd']

filter 可以为我提供实际对象的可迭代对象，但我不知道如何修改它们指向的元素。

>>> v = ['a' , 'b' , 'c' , 'd' , 'f' , 'k' , 'g' , 'a', 'd']
>>> t = ['a', 'k']
>>> p = filter(lambda x: x in t, v)
>>>
>>> for elem in p :
...    print(elem)
...
a
k
a
>>>
>>> for elem in p :
...    p = 'z'
...
>>>
>>> for elem in p :
...    print(elem)
...
>>>

Answer 1

您可以使用 map 将 v 中的值映射到固定值 z 如果它们也在 t 中> 否则保持不变:

>>> print list(map(lambda i: 'z' if i in t else i, v))
['z', 'b', 'c', 'd', 'f', 'z', 'g', 'z', 'd']

或者，您仍然可以使用推导式:

>>> v[:] = [i if not (i in t) else 'z' for i in v]
>>> print v
['z', 'b', 'c', 'd', 'f', 'z', 'g', 'z', 'd']

就我个人而言，我会发现定义函数更为明显，尤其是当您使用带有多个参数的函数时。因为通过 functools.partial 很容易“卡住”固定参数:

>>> def f(value, lookup, mapping_value):
...     if value in lookup:
...         return mapping_value
...     return value

--

>>> from functools import partial

>>> v = ['a' , 'b' , 'c' , 'd' , 'f' , 'k' , 'g' , 'a', 'd']
>>> t = ['a', 'k']

>>> func = partial(f, lookup=t, mapping_value='z')
>>> v[:] = [func(i) for i in v]
>>> print v
['z', 'b', 'c', 'd', 'f', 'z', 'g', 'z', 'd']