numpy: np.where

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]

---

---