使用NumPy庫中的unique函數(shù)可以刪除重復(fù)元素。該函數(shù)會返回一個包含唯一元素的數(shù)組，并可選擇保留第一次出現(xiàn)的元素或者保留最后一次出現(xiàn)的元素。
例如，可以這樣刪除重復(fù)元素：
import numpy as np
arr = np.array([1, 2, 2, 3, 3, 4, 5])
unique_arr = np.unique(arr)
print(unique_arr)
輸出結(jié)果為：[1 2 3 4 5]
如果想要保留最后一次出現(xiàn)的元素，可以使用參數(shù)return_index=True：
unique_arr, unique_index = np.unique(arr, return_index=True)
print(unique_arr)
print(unique_index)
輸出結(jié)果為： [1 2 3 4 5] [0 1 3 5 6]
其中，unique_index保存了唯一元素在原數(shù)組中的索引位置。
億速云公眾號
手機(jī)網(wǎng)站二維碼