查找二维数组中的元素

时间:2014-11-09 20:22:22

标签: python numpy

我正在尝试找出二维数组中的元素,例如下面的内容:

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

是否有能为我这样做的功能?

1 个答案:

答案 0 :(得分:4)

您可以使用np.unique

>>> a = np.array([[1,0,0],[1,3,0],[2,7,4]])
>>> np.unique(a)
array([0, 1, 2, 3, 4, 7])