python二维数组字典

Question

我想从一个对称矩阵的下三角形创建一个字典，表示为2d数组。例如，如果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。

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

一种方法是使用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，这最后一部分可能是：

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