np.where(condition,x,y)会返回一个数组array或元素为array的tuple元组
一、只有一个参数condition
返回一个数组中满足条件的元素的索引,且返回的是一个tuple类型,每一个元素为array类型,并对应某一维度上的索引。
二、有三个参数
满足condition的选择为x(true),不满足的选择为y(false)。
在使用三个参数的时候,要注意,condition、x、y必须具有相同的维度或者是可以广播成相同的形状,否则会报错,它返回的结果是一个列表,同原始的condition具有相同的维度和形状。
它的本质上就是选择操作,但是如果我们自己编写条件运算,使用if-else或者是列表表达式这样的语句,效率低下,故而推荐使用np.where。