Python--numpy和list，查询指定元素的个数及其位置

numpy

查询指定元素个数：
调用numpy的sum函数

>>> import numpy as np
>>> a = np.array([[0, 1, 2], [3, 4, 1]])
>>> print a
array([[0, 1, 2],
       [3, 4, 1]])
>>> print np.sum(a == 1)
2

定位指定元素：
调用numpy的where函数

>>> import numpy as np
>>> a = np.array([[0, 1, 2], [3, 4, 1]])
>>> print a
array([[0, 1, 2],
       [3, 4, 1]])
>>> print np.where(a == 1)
(array([0, 1]), array([1, 2]))

---

list

查询指定元素个数：
调用list.count(obj)函数，返回obj在list中的个数

>>> a = [0, 1, 1, 7, 5, 9]
>>> print a
[0, 1, 1, 7, 5, 9]
>>> print a.count(1)
2

定位指定元素：
调用list.index(obj)函数，返回待查找对象第一个匹配项的位置

>>> a = [0, 1, 1, 7, 5, 9]
>>> print a
[0, 1, 1, 7, 5, 9]
>>> print a.index(9)
5