python 2d配列からdict

Question

2次元配列として表現された対称行列の下三角からdictを作成したいと思います。例えば、numpy配列が;

array([[0, 2, 3], 
     [2, 0, 4], 
     [3, 4, 0]]) 

次に、私は辞書のように見えます。

{('1', '0'): 2, ('2', '0'): 3, ('2', '1'): 4} 

ベクターにも同様のポストがあります。

Fastest way to convert a Numpy array into a sparse dictionary?

私はとても感謝任意のヘルプ/ suggestoinsをPYTHONするのは比較的新しいです。

Answer 1

>>> arr =[[0, 2, 3], 
      [2, 0, 4], 
      [3, 4, 0]] 
>>> dict(((j,i), arr[i][j]) for i in range(len(arr)) for j in range(len(arr[0])) if i<j) 
{(2, 0): 3, (1, 0): 2, (2, 1): 4} 

Answer 2

これを行う1つの方法は、ndenumerateとdefaultdictです。

すべての位置に、それぞれの値をマッピングする辞書の構築：

>>> d = defaultdict(list) 
>>> for pos,val in numpy.ndenumerate(a): 
...  if val: 
...   d[val].append(pos[1]) 
... 
>>> d 
defaultdict(<class 'list'>, {2: [1, 0], 3: [2, 0], 4: [2, 1]}) 

をそして、キーと値を反転：

>>> {tuple(v):k for k,v in d.items()} 
{(2, 0): 3, (1, 0): 2, (2, 1): 4} 

あなたのPythonのバージョンは、辞書のcomprhensionをサポートしていない場合は、この最後の部分ができbe：

>>> dict((tuple(v),k) for k,v in d.iteritems()) 
{(2, 0): 3, (1, 0): 2, (2, 1): 4}