- 使用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)