arr1=np.array([1,2,2,8,0,9,6,0])
1、返回不为0元素的下标:
print(np.where(arr1))
(array([0, 1, 2, 3, 5, 6], dtype=int64),)
2、print(np.where(arr1%2==0))
(array([1, 2, 3, 4, 6, 7], dtype=int64),)
3、print(np.where(arr1%3==0,1,-1))
print(np.where(arr1%3==0,arr1,-arr1))
[-1 -1 -1 -1 1 1 1 1]
[-1 -2 -2 -8 0 9 6 0]