python - 如何仅将函数应用于某些数组元素？

Question

我有一个数组 x，我想将函数 f 应用于矩阵中满足某些条件的每个项目。 Numpy 是否提供一种机制来简化此操作？

这是一个例子。我的矩阵 x 应该只包含排他范围 (0, 1) 中的元素。但是，由于舍入误差，某些元素可能等于 0 或 1。对于 x 中恰好为 0 的每个元素，我想添加 epsilon 并为恰好为 1 的每个元素添加我想减去 epsilon。

编辑:(此编辑是在我接受 askewchan's answer 之后进行的。)另一种方法是使用 numpy.clip。 .

Answer 1

你可以这样做:

a = np.array([0,.1,.5,1])
epsilon = 1e-5
a[a==0] += epsilon
a[a==1] += -epsilon

这样做的原因是 a==0 返回一个 bool 数组，就像 Валера Горбунов 一样在他们的回答中提到:

In : a==0
Out: array([True, False, False, False], dtype=bool)

然后您将该数组用作 a 的索引，它公开了 True 而不是 False 的元素。您可以用它做很多事情，请参阅 http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html