python - 使用创建的每一行的频率从 numpy 中的矩阵中提取唯一行

Question

后续问题:

How can I use the unique(a, 'rows') from MATLAB in Python?

那里的答案解释了如何获得唯一行。然而，matlab 还会返回创建的每一行的频率。有什么优雅的方法可以用 python 实现吗？

谢谢!

Answer 1

您可以使用花式索引和评估条件来计算每个唯一行的数量，例如:

from numpy import unique, array, all
def myunique(input):
    u = array([array(x) for x in set(tuple(x) for x in input)])
    return u, array([len(input[all(input==x, axis=1)]) for x in u],dtype=int)

例子:

a = array([list('1234'),
           list('1234'),
           list('1222'),
           list('1222'),
           list('1234')],dtype=str)

print myunique(a)
#(array([['1', '2', '2', '2'],
#        ['1', '2', '3', '4']],
#       dtype='|S1'), array([2, 3]))