Numpy 数组仅将函数应用于某些元素

Question

例如，我有一个 numpy 数组，假设它有一个形状 (10,10)。现在我想将 np.exp() 应用到这个数组，但只是一些满足条件的特定元素。例如，我想将 np.exp 应用于所有不是 0 或 1 的元素。有没有办法在不使用 for 循环遍历数组的每个元素的情况下做到这一点？

Answer 1

这可以通过基本的 numpy 操作来实现。这是一种方法：

A = np.random.randint(0,5,size=(10,10)).astype(float)  # data
goods = (A!=0) & (A!=1)  # 10 x 10 boolean array
A[goods] = np.exp(A[goods])  # boolean indexing