查找二维数组中的元素

Question

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

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]

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

Answer 1

您可以使用np.unique：

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