Python NumPy unique方法用法及代碼示例

例子

基本用法

一維案例

要查找一維數組中的所有唯一值：

a = np.array([4,5,6,5])
np.unique(a)



array([4, 5, 6])

2D案例

考慮以下二維數組：

a = np.array([[4,5],[6,5]])
a



array([[4, 5],
       [6, 5]])

要查找二維數組中的所有唯一值：

np.unique(a)



array([4, 5, 6])

獲取唯一值的索引

要獲取唯一值的索引：

a = np.array([4,5,6,5])
arr_unique_values, arr_index = np.unique(a, return_index=True)
print(arr_unique_values)
print(arr_index)



[4 5 6]
[0 1 2]

獲取逆索引

獲取逆索引：

a = np.array([4,5,6,5])
arr_unique_values, arr_inverse = np.unique(a, return_inverse=True)
print(arr_unique_values)
print(arr_inverse)



[4 5 6]
[0 1 2 1]

這裏，可以使用逆來重建原始值：

arr_unique_values[arr_inverse]



array([4, 5, 6, 5])

獲取計數

要獲取計數，請設置 return_counts=True ：

a = np.array([4,5,6,5])
arr_unique_values, arr_counts = np.unique(a, return_counts=True)
print(arr_unique_values)
print(arr_counts)



[4 5 6]
[1 2 1]

在這裏，我們看到值 5 在原始數組中出現了兩次。

查找唯一的行

考慮以下二維數組：

a = np.array([[4,5],[6,5],[4,5]])
a



array([[4, 5],
       [6, 5],
       [4, 5]])

在這裏，我們看到索引 0 和索引 2 處的行是重複的。要獲取所有唯一的行：

np.unique(a, axis=0)



array([[4, 5],
       [6, 5]])

尋找獨特的列

考慮以下二維數組：

a = np.array([[4,5,4],[6,8,6]])
a



array([[4, 5, 4],
       [6, 8, 6]])

在這裏，我們看到索引 0 和索引 2 處的列是重複的。要獲取所有唯一列：

np.unique(a, axis=1)



array([[4, 5],
       [6, 8]])