np.unique

狼啸风云

发布于 2022-05-09 09:35:24

3280

发布于 2022-05-09 09:35:24

去除数组中的重复数字，并进行排序之后输出。

要求是 arr 里面是相同的维数。

import numpy as np

result1 = np.unique([1, 1, 2, 2, 2, 3, 3, 4])

print(type(result1))  # <class 'numpy.ndarray'>
print(result1)  # [1 2 3 4]

arr = np.array([[1, 2], [3, 3]])
result2 = np.unique(arr)
print(type(result2))  # <class 'numpy.ndarray'>
print(result2)  # [1 2 3]

arr = np.array([[7, 8], [3, 3], [5, 4]])
result3 = np.unique(arr)
print(type(result3))  # <class 'numpy.ndarray'>
print(result3)  # [3 4 5 7 8]

如果使用不同的维数：

依旧是排序，但是排序的对象就不同了。

arr = np.array([[7, 8], [3, 3], [5, 4, 9, 0]])
result3 = np.unique(arr)
print(type(result3))  # <class 'numpy.ndarray'>
print(result3)  # [list([3, 3]) list([5, 4, 9, 0]) list([7, 8])]

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022-03-18，如有侵权请联系 cloudcommunity@tencent.com 删除

对象