温馨提示×

NumPy数组去重的方法有哪些

小亿
105
2024-05-13 13:00:11
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

  1. 使用np.unique()函数:该函数返回输入数组中唯一值组成的数组,并可以选择返回唯一值的索引。
import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values = np.unique(arr)
print(unique_values)
  1. 使用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)
  1. 使用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元/月。点击查看>>

推荐阅读:php数组去重的方法有哪些

0