查找唯一的列和列成员

Question

他们都讨论了几种计算具有唯一行和列的矩阵的方法。

但是，至少对于未经训练的人来说，解决方案看起来有些复杂。例如，这是第一个线程的最佳解决方案（如果我错了，请纠正我），我认为这是最安全，最快的：

np.unique(a.view(np.dtype((np.void, a.dtype.itemsize*a.shape[1])))).view(a.dtype).reshape(-1, 
a.shape[1])

无论哪种方式，上述解决方案都只返回唯一行的矩阵。我正在寻找的是具有原始功能的东西np.unique

u, indices = np.unique(a, return_inverse=True)

它不仅返回唯一条目的列表，而且还返回找到的每个唯一条目的每个项目的成员资格，但是我该如何对列执行此操作？

这是我要寻找的示例：

array([[0, 2, 0, 2, 2, 0, 2, 1, 1, 2],
       [0, 1, 0, 1, 1, 1, 2, 2, 2, 2]])

我们会有：

u       = array([0,1,2,3,4])
indices = array([0,1,0,1,1,3,4,4,3])

其中的不同值u表示原始数组中唯一列的集合：

0 -> [0,0]
1 -> [2,1]
2 -> [0,1]
3 -> [2,2]
4 -> [1,2]

Answer 1

本质上，您希望np.unique返回唯一列的索引以及它们的使用位置的索引？通过转置矩阵，然后使用另一个问题的代码并添加，这很容易做到return_inverse=True。

at = a.T
b = np.ascontiguousarray(at).view(np.dtype((np.void, at.dtype.itemsize * at.shape[1])))
_, u, indices = np.unique(b, return_index=True, return_inverse=True)

使用a，可以得到：

In [35]: u
Out[35]: array([0, 5, 7, 1, 6])

In [36]: indices
Out[36]: array([0, 3, 0, 3, 3, 1, 4, 2, 2, 4])

但是，我对您想u成为的人并不完全清楚。如果希望它成为唯一列，则可以使用以下内容：

at = a.T
b = np.ascontiguousarray(at).view(np.dtype((np.void, at.dtype.itemsize * at.shape[1])))
_, idx, indices = np.unique(b, return_index=True, return_inverse=True)
u = a[:,idx]

这会给

In [41]: u
Out[41]:
array([[0, 0, 1, 2, 2],
       [0, 1, 2, 1, 2]])

In [42]: indices
Out[42]: array([0, 3, 0, 3, 3, 1, 4, 2, 2, 4])