用法一:可以返回一个n维数组,可广播。
# np.where(condition, x, y)
# condition, x, y, return are all can be #ndarray
x = np.random.randn((4, 4))
np.where(x>0, 2, -2)
output:
用法二:condition 也可以是布尔型数组,每个条件都和x,y对应(广播)
用法三:找到n维数组中特定数值的索引
用法一:可以返回一个n维数组,可广播。
# np.where(condition, x, y)
# condition, x, y, return are all can be #ndarray
x = np.random.randn((4, 4))
np.where(x>0, 2, -2)
output:
用法二:condition 也可以是布尔型数组,每个条件都和x,y对应(广播)
用法三:找到n维数组中特定数值的索引