Comparing two NumPy arrays for equality, element-wise

by 
(A==B).all()

test if all values of array (A==B) are True.

Note: maybe you also want to test A and B shape, such as A.shape == B.shape

Special cases and alternatives (from dbaupp’s answer and yoavram’s comment)

It should be noted that:

  • this solution can have a strange behavior in a particular case: if either A or B is empty and the other one contains a single element, then it return True. For some reason, the comparison A==B returns an empty array, for which the all operator returns True.
  • Another risk is if A and B don’t have the same shape and aren’t broadcastable, then this approach will raise an error.

In conclusion, if you have a doubt about A and B shape or simply want to be safe: use one of the specialized functions:

np.array_equal(A,B)  # test if same shape, same elements values
np.array_equiv(A,B)  # test if broadcastable shape, same elements values
np.allclose(A,B,...) # test if same shape, elements have close enough values

More Related Contents:

  1. Element-wise string concatenation in numpy
  2. What are the advantages of NumPy over regular Python lists?
  3. ValueError: setting an array element with a sequence
  4. Numpy `logical_or` for more than two arguments
  5. Replace all elements of Python NumPy Array that are greater than some value
  6. Finding indices of matches of one array in another array
  7. Why are NumPy arrays so fast?
  8. Calculate mean across dimension in a 2D array
  9. How to copy a 2D array into a 3rd dimension, N times?
  10. How do I convert a numpy array to (and display) an image?
  11. Numpy random choice to produce a 2D-array with all unique values
  12. How do I print the full NumPy array, without truncation?
  13. numpy array assignment problem
  14. Preserve custom attributes when pickling subclass of numpy array
  15. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  16. Why is a `for` over a Python list faster than over a Numpy array?
  17. Subtracting numpy arrays of different shape efficiently
  18. Input and output numpy arrays to h5py
  19. Why does numpy have a corresponding function for many ndarray methods?
  20. Numpy sub-array assignment with advanced, mixed indexing
  21. Setting the fmt option in numpy.savetxt
  22. Access n-th dimension in python [duplicate]
  23. numpy division with RuntimeWarning: invalid value encountered in double_scalars
  24. How does the axis parameter from NumPy work?
  25. numpy – evaluate function on a grid of points
  26. numpy arange: how to make “precise” array of floats?
  27. Shift rows of a numpy array independently
  28. Interweaving two NumPy arrays efficiently
  29. Formatting floats in a numpy array [duplicate]
  30. How to convert a 3d numpy array to 2d

Leave a Comment