У меня есть следующий словарь:
g = {
'A': ['A', 'B', 'C'],
'B': ['A', 'C', 'E'],
'C': ['A', 'B', 'D'],
'D': ['C','E'],
'E': ['B','D']
}
Он реализует граф, каждый список содержит соседей вершин графа (ключи словаря — это сами вершины). У меня проблемы, я не могу придумать способ получить матрицу смежности графа из своих списков соседей, может быть, это легко, но я новичок в python, надеюсь, кто-нибудь может мне помочь! Я использую Python 3.5
Мне нужно создать следующую матрицу:
без панд
keys=sorted(g.keys())
size=len(keys)
M = [ [0]*size for i in range(size) ]
for a,b in [(keys.index(a), keys.index(b)) for a, row in g.items() for b in row]:
M[a][b] = 2 if (a==b) else 1
M
[2, 1, 1, 0, 0],
[1, 0, 1, 0, 1],
[1, 1, 0, 1, 0],
[0, 0, 1, 0, 1],
[0, 1, 0, 1, 0]]
Пояснение
for a, row in g.items() перебирает записи ключ:значение в словаре, а for b in row перебирает значения. Если бы мы использовали (a,b), это дало бы нам все пары.
(keys.index(a), keys.index(b)) Но нам нужен индекс для присвоения соответствующей записи матрицы,
keys=sorted(g.keys()) вот почему мы извлекли и отсортировали ключи.
for a,b in... получение записей индекса и присвоение значения 1 или 2 на основе диагонального элемента или нет.
Матрица M = [ [0]*size for ... не может использоваться до инициализации.
Вот решение с использованием pandas.
import pandas as pd
g = {
'A': [ 'A', 'B', 'C'],
'B': [ 'A', 'C', 'E'],
'C': [ 'A', 'B ',' D '], # I added a comma here
'D': [' C ',' E '],
'E': [' B ',' D ']
}
# clean up the example
g = {k: [v.strip() for v in vs] for k, vs in g.items()}
edges = [(a, b) for a, bs in g.items() for b in bs]
df = pd.DataFrame(edges)
adj_matrix = pd.crosstab(df[0], df[1])
# 1 A B C D E
# 0
# A 1 1 1 0 0
# B 1 0 1 0 1
# C 1 1 0 1 0
# D 0 0 1 0 1
# E 0 1 0 1 0
Я не уверен, почему у вас есть 2 в матрице вашего примера в позиции (A, A).
