python - 如何在 numpy 中创建一个分量大于数字的向量？

Question

这个问题已经有答案了:

How can I replace values less than or equal to 0 with 1 in a 2d numpyarray? (3 个回答)

Forcing Elements in a Numpy Array to be Within a Specified Range [duplicate] (1 个回答)

已关闭 5 年前。

例如，我有一个像这样的向量:

v = np.array([2, 4, 1, 7, 9, 3, 5])

我想将每个分量与一个数字进行比较，以获得一个新的向量，其中哪些分量大于或等于该数字(如果原始分量较小)。对于上面的例子和limit=3结果应该是:

np.array([3, 4, 3, 7, 9, 3, 5])

我该如何使用 numpy 来做到这一点？

Answer 1

你可以简单地做

v[v <= 3] = 3

这会为您提供所需的输出:

array([3, 4, 3, 7, 9, 3, 5])

简要说明:

v <= 3

创建一个 bool 数组

array([ True, False,  True, False, False,  True, False], dtype=bool)

和命令

v[v <= 3]

将所有值设置为 3，其中此 bool 数组的计算结果为 True。