numpy使用where函数定位满足条件的下标