NumPy如何删除重复元素

使用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保存了唯一元素在原数组中的索引位置。