Сравнение нескольких массивов numpy

Question

Как сравнить более 2 массивов numpy?

import numpy 
a = numpy.zeros((512,512,3),dtype=numpy.uint8) 
b = numpy.zeros((512,512,3),dtype=numpy.uint8) 
c = numpy.zeros((512,512,3),dtype=numpy.uint8) 
if (a==b==c).all(): 
    pass 

это дает ValueError, и я не заинтересован в сравнении массивов двух одновременно.

Answer 1

Для трех массивов вы можете проверить равенство между соответствующими элементами между первым и вторым массивами, а затем вторым и третьим массивами, чтобы дать нам два булевых скаляра и, наконец, увидеть, являются ли оба этих скаляра True для окончательного скалярного вывода, как так -

np.logical_and((a==b).all(), (b==c).all()) 

для большего числа массивов, вы можете складывать их, получить дифференцирование вдоль оси укладки и проверьте все этих дифференцировки равны нулям. Если они есть, мы имеем равенство среди всех входных массивов, иначе нет. Реализация будет выглядеть так -

L = [a,b,c] # List of input arrays 
out = (np.diff(np.vstack(L).reshape(len(L),-1),axis=0)==0).all() 

Answer 2

Для три массива, вы должны действительно просто сравнить их два, в то время:

if np.array_equal(a, b) and np.array_equal(b, c): 
    do_whatever() 

Для переменного числа массивов, давайте предположим, что все они объединены в один большой массив arrays. Тогда вы могли бы сделать

if np.all(arrays[:-1] == arrays[1:]): 
    do_whatever()