MATLAB 函数的 Python 等效项 unique

Question

我知道有人问过这个问题，但我还没有找到答案。非常感谢任何帮助

在 Matlab 中写为：[C,ia,ic] = unique(A)

我对所有输出元素感兴趣，即C、ia 和 ic

这是 matlab 函数作用的示例

A = [9 2 9 5];
Find the unique values of A and the index vectors ia and ic, 
such that C = A(ia) and A = C(ic).

[C, ia, ic] = unique(A)
C = 1×3

     2     5     9

ia = 3×1

     2
     4
     1

ic = 4×1

     3
     1
     3
     2

请问如何在 python 中重现这个？如前所述，我对所有输出元素感兴趣，即C、ia 和 ic

Answer 1

使用numpy.unique的解决方案（感谢@SBad本人对解决方案质量的提升）：

import numpy as np

A = np.array([9,2,9,5])

C, ia, ic = np.unique(A, return_index=True, return_inverse=True)

print(C)
print(ia)
print(ic)

输出

[2 5 9]
[1 3 0]
[2 0 2 1]

---

通过列表理解，您还可以获得ic：

ic = [i for j in A for i,x in enumerate(C) if x == j]

---

注意：

请记住，MATLAB 使用基于 1（一）的索引，而 Python 使用基于 0（零）的索引。