从矩阵蟒蛇形成一个边列表

Question

我有一个矩阵，如下所示：

Matrix = [[0,0,1], 
      [1,0,1], 
      [0,1,0]] 

我想形成一个边列表看起来像以下：

Edge_list = {0: [2], 
      1: [0, 2], 
      2: [1]} 

的Edge_list中的第一列表示节点，而下一列表示其具有边缘的节点（矩阵值为1）。

任何帮助将不胜感激。谢谢。

Answer 1

使用dict comprehension和list comprehension与enumerate：

>>> matrix = [ 
...  [0,0,1], 
...  [1,0,1], 
...  [0,1,0], 
... ] 
>>> 
>>> {i: [j for j, x in enumerate(row) if x] for i, row in enumerate(matrix)} 
{0: [2], 1: [0, 2], 2: [1]}