python：用条件替换列表中的元素

Question

我正在尝试使用 python 执行以下操作，但我有一个奇怪的行为。假设我有以下列表：

x = [5, 4, 3, 2, 1]

现在，我正在做类似的事情：

x[x >= 3] = 3

这给出了：

x = [5, 3, 3, 2, 1]

为什么只有第二个元素被改变？我期待：

[3, 3, 3, 2, 1]

Answer 1

因为 Python 会将x>=3 评估为True，并且由于True 等于1，所以x 的第二个元素将被转换为3。

为此，您需要使用列表理解：

>>> [3 if i >=3 else i for i in x]
[3, 3, 3, 2, 1]

如果您想知道为什么x >= 3 评估为 True，请参阅以下documentation：

CPython 实现细节：除数字外的不同类型的对象按其类型名称排序；不支持正确比较的相同类型的对象按其地址排序。

当然，在 python-2.x 和 CPython 实现中，列表总是大于整数类型。因为字符串大于列表：

>>> ''>[]
True

但是，在 Python-3.X 中，您不能将不可排序的类型一起比较，结果中会得到 TypeError。

In [17]: '' > []
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-17-052e7eb2f6e9> in <module>()
----> 1 '' > []

TypeError: unorderable types: str() > list()

Answer 2

您可以将此语法与Numpy 一起使用：

>>> import numpy as np
>>> x = np.array([5, 4, 3, 2, 1])
>>> x[x>3]=3
>>> x
array([3, 3, 3, 2, 1])

您也可以使用Pandas：

>>> import pandas as pd
>>> x = pd.Series([5, 4, 3, 2, 1])
>>> x
0    5
1    4
2    3
3    2
4    1
dtype: int64
>>> x[x>3]=3
>>> x
0    3
1    3
2    3
3    2
4    1
dtype: int64

Answer 3

您正在使用 python 列表。在 python(2.x) 中，list 与 int 的比较将比较 类型，而不是值。因此，您的比较结果为True，它等同于1。换句话说，你的表达式相当于：

x[1] = 3  # x[1] == x[True] == x[x > 3]

注意，python3.x 不允许这种类型的比较（因为它几乎肯定不是你的意思）——如果你想做这种操作，你几乎肯定会通过查看 numpy 文档想到它因为 numpy API 是专门为支持这种事情而设计的：

import numpy as np
array = np.arange(5)
array[array > 3] = 3