Python numpy数组中值的替换

例1.置换数组中的0值。

输入：

import numpy as np

a = np.array([[1,2,0],[4,0,1],[0,5,0]])
a[a==0] = 1
print(a)

输出：

[[1 2 1]
 [4 1 1]
 [1 5 1]]

例2.置换数组中的负值。

输入：

import numpy as np

b = np.array([[-1,2,0],[-4,0,1],[0,-5,0]])
b[b<0] = 0
print(b)

输出：

[[0 2 0]
 [0 0 1]
 [0 0 0]]

例3.在置换数组中的正值。

输入：

import numpy as np

c = np.array([[1,2,0],[4,0,1],[0,5,0]])
c[c>0] = -1
print(c)

输出：

[[-1 -1  0]
 [-1  0 -1]
 [ 0 -1  0]]