Finding what elements are in a 2-D array

Question

I'm trying to find what elements are in a 2-D array, such as something along the lines below:

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]

Is there a function that would do this for me?

DSM · Accepted Answer · 2014-11-09 20:23:53Z

4

You could use np.unique:

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

answered Nov 9, 2014 at 20:23

DSM

355k67 gold badges606 silver badges504 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Thank you very much m(_ _)m The np.unique function is just what I was looking for.