Sometimes, we want to test whether a Python NumPy array contains a given row.

In this article, we’ll look at how to test whether a Python NumPy array contains a given row.

How to test whether a Python NumPy array contains a given row?

To test whether a Python NumPy array contains a given row, we can convert the NumPy array to a list and then use the in operator to check whether the list is in the nested list.

For instance, we write:

import numpy as np

a = np.array([[1, 2], [10, 20], [100, 200]])
l = a.tolist()
print([1, 2] in l)
print([1, 200] in l)

We can np.array with a nested list to create an array.

Then we call a.tolist to convert the NumPy array back to a list.

Next, we use the in operator to check whether [1, 2] and [1, 200] is in l.

Therefore, print should print:

True
False

since [1, 2] is in l and [1, 200] isn’t.

Conclusion

To test whether a Python NumPy array contains a given row, we can convert the NumPy array to a list and then use the in operator to check whether the list is in the nested list.