numpy

NumPy如何删除重复元素

小亿
150
2024-05-13 12:51:12
栏目: 编程语言

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

0
看了该问题的人还看了