Syntax
numpy.where(condition[, x, y])
返回 符合某一条件 的 下标 。
Note :
不接受 list
型的参数,只接受 `ndarray
型输入。
用法
单目运算
import numpy as np
array_1 = np.array([1, 0, 1, 5, 5, 1])
array_2 = np.array([[1, 0, 1],
[5, 5, 1]])
print np.where(array_1 > 2)
print np.where(array_2 > 2)
打印结果:
(array([3, 4]),)
(array([1, 1]), array([0, 1]))
三目运算
例:将奇数转换为偶数,偶数转换为奇数:
import numpy as np
y = np.array([1, 2, 3, 4, 5, 6])
print np.where(y%2 == 0, y+1, y-1)
打印结果:
[0 3 2 5 4 7]