温馨提示×

NumPy数组去重的方法有哪些

小亿
91
2024-05-13 13:00:11
栏目: 编程语言

  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)

0