python----np.where()

np.where(condition[, x, y])

contidion----条件

x----可选参数，与condition的维度相同，condition满足为True时，返回x

y----可选参数，与condition的维度相同，不满足为False时，返回y