Pythonの配列の一意性チェック

Question

私はnumpy配列のリストを持っています。各配列は、バイナリ値のシーケンスである。 [0,1,1,0,1,0]。私はリスト内のすべての配列を比較し、全く同じシーケンスの配列を持つ配列のすべてのインデックスを持つ別のリストを取得したいと思います。

たとえば、A=[[0,1,0],[0,0,1],[0,1,0],[1,0,0],[1,0,0]]の場合、B = [[1,3],[2],[4,5]]のようなものを取得したいとします。

私は次のようにループのための二重を使用して一意性または各配列の類似性を確認することができる午前：

for i in xrange(len(A)): 
    for j in xrange(i+1,len(A)): 
     if (A[i]==A[j]).all(): 
      print 'Duplicate of arrays %d and %d' %(i,j) 

が、その後、私はどのようにグループにして、それらの間に等しい配列のインデックスを知りません別のリスト、配列、行列、または何でも。

アドバイスをいただき、ありがとうございます。

あなたは、単に辞書のアプローチを使用することができ

Answer 1

：インデックスが0とここに始めること

dic = {} 
B = [] 
for idx,row in enumerate(A): 
    trow = tuple(row) 
    if trow not in dic: 
     dic[trow] = len(dic) 
     B.append([idx]) 
    else: 
     B[dic[trow]].append(idx) 

注意、あなたはインデックスは1で開始したい場合、あなたは、単にそれを供給することによって、コードを変更し使用することができます

dic = {} 
B = [] 
for idx,row in enumerate(A,1): 
    trow = tuple(row) 
    if trow not in dic: 
     dic[trow] = len(dic) 
     B.append([idx]) 
    else: 
     B[dic[trow]].append(idx)

resultとして生成します：

コールへ

$ python3 
Python 3.5.2 (default, Nov 17 2016, 17:05:23) 
[GCC 5.4.0 20160609] on linux 
Type "help", "copyright", "credits" or "license" for more information. 
>>> A=[[0,1,0],[0,0,1],[0,1,0],[1,0,0],[1,0,0]] 
>>> dic = {} 
>>> B = [] 
>>> for idx,row in enumerate(A,1): 
...  trow = tuple(row) 
...  if trow not in dic: 
...   dic[trow] = len(dic) 
...   B.append([idx]) 
...  else: 
...   B[dic[trow]].append(idx) 
... 
>>> B 
[[1, 3], [2], [4, 5]] 

Answer 2

import numpy_indexed as npi 
npi.group_by(A).split(np.arange(len(A)))