NumPy如何删除重复元素

   2024-10-18 6160
核心提示:使用NumPy库中的unique函数可以删除重复元素。该函数会返回一个包含唯一元素的数组,并可选择保留第一次出现的元素或者保留最后

使用NumPy库中的unique函数可以删除重复元素。该函数会返回一个包含唯一元素的数组,并可选择保留第一次出现的元素或者保留最后一次出现的元素。

例如,可以这样删除重复元素:

import numpy as nparr = 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保存了唯一元素在原数组中的索引位置。

 
举报打赏
 
更多>同类维修大全
推荐图文
推荐维修大全
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号