1、numpy.where
numpy.where的参数:
numpy.where(condition,x,y),即:条件满足时值为x,条件不满足时值为y,condition必须是元素为bool类型的array(数组)类型。
2、numpy.piecewise
numpy.piecewise的参数:
numpy.piecewise(array,[condition1,condition2],[value1,value2]),即:传入一个array,condition1满足时值为value1,condition2满足时值为value2;value1和value2也可传入函数[function1,condition2],即array中的元素满足condition1时执行function1,满足condition2时执行function2。
可用于对数据进行批量化处理。