使用NumPy库中的unique函数可以删除重复元素。该函数会返回一个包含唯一元素的数组,并可选择保留第一次出现的元素或者保留最后一次出现的元素。
例如,可以这样删除重复元素:
import numpy as np
arr = np.array([1, 2, 2, 3, 3, 4, 5])
unique_arr = np.unique(arr)
print(unique_arr)
输出结果为:[1 2 3 4 5]
如果想要保留最后一次出现的元素,可以使用参数return_index=True:
unique_arr, unique_index = np.unique(arr, return_index=True)
print(unique_arr)
print(unique_index)
输出结果为: [1 2 3 4 5] [0 1 3 5 6]
其中,unique_index保存了唯一元素在原数组中的索引位置。