Matlab中唯一函数的Python等价物

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]))