技術標籤：python科學計算pythonnumpy

不知道大家有沒有過這種經歷，就是想要判斷兩個陣列運算後得到的新陣列中的各個元素值是否相同。這裡給出一種使用np.unique()的方法，程式碼如下：

import numpy as np


class Debug:
    @staticmethod
    def isAllElementSame():
        x1 = np.array([[1, 2, 3], [3, 4, 5], [6, 7, 8]])
        x2 = np.array([[81., 162., 243., ], [243., 324., 405
 
.], [486., 567., 648.]])
        print('The result if x2/x1 is:')
        print(x2 / x1)
        print('Judge whether all elements in array are same or not')
        print(len(np.unique(x2 / x1)) == 1)


if __name__ == '__main__':
    debug = Debug()
    debug.isAllElementSame()
"""
The result if x2/x1 is:
[[81. 81. 81.]
 [81. 81. 81.]
 [81. 81. 81.]]
Judge whether all elements in array are same or not
True
"""
 

可以看到，當輸出為True的時候，表明陣列中的所有元素的值均一致，反之，當為False的時候，陣列中存在不一樣的元素值。