numpy中where函数的用法
numpy.
where(condition,x,y)——若满足condition,输出x,否则输出y。
举例:
例1:一维数组
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a < 5, a, 10*a)
array([ 0, 1, 2, 3, 4, 50, 60, 70, 80, 90])
例2:多维数组
>>> np.where([[True, False], [True, True]],
... [[1, 2], [3, 4]],
... [[9, 8], [7, 6]])
array([[1, 8],
[3, 4]])
这个例子解释一下:条件是
[True, False], [True, True]
也就是说,第一个元素认为是true(真的),则应选择真值表 [1, 2], [3, 4] 中对应的第一个元素——1。第二个元素认为是false(假的),则应选择假值表[9,