NumPy All: Understanding np.all()

The np.all() function tests whether all elements in a NumPy array evaluate to true:

np.all(np.array([[1, 1], [1, 1]]))

# Expected result
# True

Notice the input can have arbitrary shape and the data type does not have to be boolean (it just has to be truthy). If any of the elements don’t evaluate to true, the function returns false:

np.all(np.array([[0, 1], [1, 1]]))

# Expected result
# False

We can also use the optional axis argument to make np.all() a reducing operation. Say we want to know which rows in a matrix have elements that all evaluate to true. We can do that by passing in axis=-1:

np.all(np.ones((2, 3)), axis=-1)

# Expected result
# array([ True,  True])

In the above example, there are two rows and for each of them, all elements evaluate to true. The -1 value here is shorthand for “the last axis”.

And that’s it! NumPy also has a function called np.any() which has the same API as np.all() but returns true when any of the elements evaluate to true.

Hello, my name is Ben Cook

I help data scientists deploy their code. If there's any way I can serve you, don't hestitate to reach out. You can also find out a little more about me or download my free guide: 8 Best Practices for Building Machine Learning Pipelines.

Thanks for stopping by!

Contact

Get in touch

Connect

Free Guide

Download my free guide: 8 Best Practices for Building Machine Learning Pipelines