список ребер питона к матрице смежности - PullRequest
0 голосов
/ 31 марта 2019

Учитывая список ребер, мне нужно преобразовать список в матрицу смежности в Python.Я очень, очень близок, но я не могу понять, что я делаю неправильно.Где мое неправильное мышление?

E= [[0, 0], [0, 1], [1, 0], [1, 1]]

nmax = max(E)
nmax2 =max(nmax)

m = []
for i in range(nmax2+1):
    row = []
    for j in range(nmax2+1):
         if [i,j]== E[i]:
               row.append(1)
         else:
               row.append(0)
    m.append(row)

 print(m)

Я ожидаю, что результат будет: 1 1 1 1

Но мой код выдает: 1 0 0 0

Ответы [ 2 ]

0 голосов
/ 31 марта 2019

я бульвар ниже, чище и делает работу

    E= [[0, 0], [0, 1], [1, 0], [1, 1]]
    size = max(max(E))+1
    r = [[0 for i in range(size)] for j in range(size)]
    for row,col in E:
        r[row][col] = 1
    print(r)    
0 голосов
/ 31 марта 2019

Как следует из комментария, вы проверяете ребра только для того количества строк, которое имеется в вашей матрице смежности, поэтому в общем случае вам не удается достичь многих ребер.Вместо этого рассмотрим следующее:

E = [[0, 0], [0, 1], [1, 0], [1, 1]]

# nodes must be numbers in a sequential range starting at 0 - so this is the
# number of nodes. you can assert this is the case as well if desired 
size = len(set([n for e in E for n in e])) 
# make an empty adjacency list  
adjacency = [[0]*size for _ in range(size)]
# populate the list for each edge
for sink, source in E:
    adjacency[sink][source] = 1

>>> print(adjacency)
>>> [[1, 1], [1, 1]]

Если вы хотите быть краткими за счет ясности:

adjacency = [[1 if [i, j] in set(map(tuple, E)) else 0 for j in range(size)] for i in range(size)]

size, представляющих количество узлов - как и раньше.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...