python查找元素

python实现数组元素的查找

对于数组我们通常需要找到特定元素的位置，本次主要用到的函数为numpy模块下的nonzero函数。
代码示例：

import numpy as np

data = np.array([[1, 2, 3, 4], [5, 6 , 7, 1]])
site = np.transpose(np.nonzero(data == 1))
print('data:\n{0}, \n sites:\n{1}'.format(data, site))

输出：

data:
[[1 2 3 4]
 [5 6 7 1]], 
 sites:
[[0 0]
 [1 3]]

代码中的data == 1 可以换为其他条件判断语句，例如：data<4，等。