溫馨提示×

NumPy數(shù)組去重的方法有哪些

小億
98
2024-05-13 13:00:11
欄目: 編程語言

  1. 使用np.unique()函數(shù)：該函數(shù)返回輸入數(shù)組中唯一值組成的數(shù)組，并可以選擇返回唯一值的索引。
import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values = np.unique(arr)
print(unique_values)
  1. 使用set()函數(shù)：將NumPy數(shù)組轉(zhuǎn)換為Python的set類型，然后再轉(zhuǎn)換回NumPy數(shù)組。
import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values = np.array(list(set(arr)))
print(unique_values)
  1. 使用np.unique()函數(shù)的return_index參數(shù)：設(shè)置return_index參數(shù)為True，可以返回唯一值在輸入數(shù)組中的索引。
import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3])
unique_values, unique_index = np.unique(arr, return_index=True)
print(unique_values)
print(unique_index)

0 踩