np.array_equal()
- 1、该方法可以检查两个数组是否具有相同的形状和元素。
- 2、该方法还可用在数组和列表之间、列表与列表间的比较
# 一维数组,或者列表
a = np.array([1, 3, 5, 7, 9])
b = np.array([2, 4, 6, 8, 0])
list_a = [1, 3, 5, 7, 9] # 列表,在python中一维数组被称为列表,numpy则叫做数组
list_b = [2, 4, 6, 8, 0]
copy_a = np.array([1, 3, 5, 7, 9])
# 不同位数的数组
c = np.array([1, 3, 5])
# 二维数组
arr1 = np.array([[1, 3, 5], [2, 4, 6]])
arr2 = np.array([[1, 3, 5], [2, 4, 6]])
arr3 = np.array([[2, 4, 6], [1, 3, 5]])
print("数组和数组比较:{}".format(np.array_equal(a, b))) # 数组和数组比较
print("数组和列表比较:{}".format(np.array_equal(a, list_a))) # 数组和列表比较
print("不同位数的数组比较:{}".format(np.array_equal(a, c))) # 不同位数的数组比较
print("二维数组间的比较:{}==={}".format(np.array_equal(arr1, arr2), np.array_equal(arr1, arr3)))
print("不同维度数组的比较:{}".format(np.array_equal(a, arr1)))
输出:
数组和数组比较:False
数组和列表比较:True
不同位数的数组比较:False
二维数组间的比较:True===False
不同维度数组的比较:False
np.unique()
a = np.array([6, 5, 7, 3, 6, 3, 4, 3, 9, 3, 0, 6])
arr_1 = np.array([[3, 2, 3, 6], [3, 0, 3, 8], [3, 2, 3, 6]])
print(a)
print(np.unique(a))
print("="*40)
print(arr_1)
print("-"*20)
print(np.unique(arr_1, axis=0))
print("-"*20)
print(np.unique(arr_1, axis=1))
print("="*40)
输出
[6 5 7 3 6 3 4 3 9 3 0 6]
[0 3 4 5 6 7 9]
========================================
[[3 2 3 6]
[3 0 3 8]
[3 2 3 6]]
--------------------
[[3 0 3 8]
[3 2 3 6]]
--------------------
[[2 3 6]
[0 3 8]
[2 3 6]]
========================================
arr_1 = np.array([[3, 0, 3, 6], [3, 2, 3, 8], [3, 0, 3, 6]])
print(arr_1)
print("-"*20)
print(np.unique(arr_1, axis=0))
print("-"*20)
print(np.unique(arr_1, axis=1))
输出:
[[3 0 3 6]
[3 2 3 8]
[3 0 3 6]]
--------------------
[[3 0 3 6]
[3 2 3 8]]
--------------------
[[0 3 6]
[2 3 8]
[0 3 6]]