温馨提示×

NumPy数组去重的方法有哪些

numpy

小亿

105

2024-05-13 13:00:11

栏目: 编程语言

开发者测试专用服务器限时活动，0元免费领，库存有限，领完即止！点击查看>>

使用np.unique()函数：该函数返回输入数组中唯一值组成的数组，并可以选择返回唯一值的索引。

import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values = np.unique(arr)
print(unique_values)

使用set()函数：将NumPy数组转换为Python的set类型，然后再转换回NumPy数组。

import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values = np.array(list(set(arr)))
print(unique_values)

使用np.unique()函数的return_index参数：设置return_index参数为True，可以返回唯一值在输入数组中的索引。

import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values, unique_index = np.unique(arr, return_index=True)
print(unique_values)
print(unique_index)

亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

0 赞

0 踩

NumPy数组去重的方法有哪些

最新问答

相关标签