【numpy】np.where() + np.unique()的使用

1、np.where()的使用

有两种用法：

• np.where(condition, x, y)：当condition为True时，返回x，否则返回y

# 例1, x,y为int值
a = np.arange(10)
# array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
print(np.where(a > 5, 1, -1))
# array([-1, -1, -1, -1, -1, -1,  1,  1,  1,  1]) 
"""
例2：x, y也可以是数组，三个维度要相同，根据condition对应位置的bool值选择x和y对应位置的值
"""
np.where([[True,False], [True,True]],   
			 [[1,2], [3,4]],
             [[9,8], [7,6]])
# 输出 array([[1, 8], [3, 4]])

• np.where(condition)：返回满足condition条件的数组的索引值，多维数组会以元组形式给出。

""" 例1：一维数组 """
a = np.array([2,4,6,8,10])
np.where(a > 5