numpy.
where
返回从中选择的元素 x 或 y 取决于 condition .
注解
只有时 condition 提供了,此函数是 np.asarray(condition).nonzero() . 使用 nonzero 直接应该首选,因为它对子类的行为是正确的。本文档的其余部分仅涵盖提供了所有三个参数的情况。
np.asarray(condition).nonzero()
nonzero
如果是真的,屈服 x ,否则屈服 y .
要从中选择的值。 x , y 和 condition 需要能够以某种形式广播。
元素来自的数组 x 在哪里? condition 是真的,元素来自 y 别处。
参见
choose
省略x和y时调用的函数
笔记
如果所有数组都是一维的, where 等于:
[xv if c else yv for c, xv, yv in zip(condition, x, y)]
实例
>>> 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])
这也可用于多维数组:
>>> np.where([[True, False], [True, True]], ... [[1, 2], [3, 4]], ... [[9, 8], [7, 6]]) array([[1, 8], [3, 4]])
x、y和条件的形状一起广播:
>>> x, y = np.ogrid[:3, :4] >>> np.where(x < y, x, 10 + y) # both x and 10+y are broadcast array([[10, 0, 0, 0], [10, 11, 1, 1], [10, 11, 12, 2]])
>>> a = np.array([[0, 1, 2], ... [0, 2, 4], ... [0, 3, 6]]) >>> np.where(a < 4, a, -1) # -1 is broadcast array([[ 0, 1, 2], [ 0, 2, -1], [ 0, 3, -1]])