Matlab中唯一函数的Python等价物

在Matlab中，unique函数用于找出数组中的唯一元素，并且可以按照不同的参数选项返回排序后的唯一值、唯一值的索引或者唯一值在原数组中的计数。在Python中，可以使用NumPy库中的numpy.unique函数来实现类似的功能。

基础概念

唯一元素：在一个集合中，不重复出现的元素。
排序：将元素按照一定的顺序排列。
索引：元素在数组中的位置。
计数：元素在数组中出现的次数。

类型

唯一值：返回数组中的不重复元素。
索引：返回唯一值在原数组中的位置。
计数：返回每个唯一值在原数组中出现的次数。

应用场景

数据清洗：去除重复的数据记录。
数据分析：统计不同类别的出现频率。
集合运算：进行集合相关的操作，如并集、交集等。

示例代码

以下是一个使用numpy.unique函数的示例：

import numpy as np

# 创建一个包含重复元素的数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])

# 返回唯一值
unique_values = np.unique(arr)
print("Unique values:", unique_values)

# 返回唯一值及其在原数组中的索引
unique_indices = np.unique(arr, return_index=True)
print("Unique values with their indices:", unique_indices)

# 返回唯一值及其在原数组中的计数
unique_counts = np.unique(arr, return_counts=True)
print("Unique values with their counts:", unique_counts)

# 返回唯一值、索引和计数
unique_all = np.unique(arr, return_index=True, return_counts=True)
print("Unique values, indices and counts:", unique_all)

输出

Unique values: [1 2 3 4]
Unique values with their indices: (array([1, 2, 3, 4]), array([0, 1, 3, 6]))
Unique values with their counts: (array([1, 2, 3, 4]), array([1, 2, 3, 4]))
Unique values, indices and counts: (array([1, 2, 3, 4]), array([0, 1, 3, 6]), array([1, 2, 3, 4]))