# numpy.where的使用
# 函数介绍
numpy.where(condition, [x, y, ])
condition: 条件x:数组，shape同condition或支持广播y:数组，shape同condition或支持广播
根据condition从x或y中取数值，如果condition条件为True取x对应位置的元素，否则取y对应位置的元素。
当数组都是一维的时候，函数的功能等同于以下语句：
[xv if c else yv for c, xv, yv in zip(condition, x, y)]
# 示例
[x,y]不为空
cond = [[True, False], [True, True]]
x = [[1, 2], [3, 4]]
y = [[9, 8], [7, 6]]
out = np.where(cond, x, y)
print(out)
# [[1, 8],
# [3, 4]]
可以看到，条件为True的位置取了x的值，为False的位置取了y的值。
[x,y]为空
cond = [[True, True, False], [True, True, True]]
x, y = np.where(cond)
print(x,y)
# [0 0 1 1 1] [0 1 0 1 2]
print(np.stack((x, y)).T)
# array([[0, 0],
# [0, 1],
# [1, 0],
# [1, 1],
# [1, 2]])
可以看到当不传入[x,y]数组时，np.where返回的时condition为True位置的i行j列的坐标值。可以看到上面condition对应的(0,0)/(0,1)/(1,0)等5个位置都是True。
1.https://numpy.org/doc/stable/reference/generated/numpy.where.html (opens new window)