Как получить имена графов после преобразования графа в смежную матрицу? - PullRequest
Новая
       16

Как получить имена графов после преобразования графа в смежную матрицу?

0 голосов
/ 25 октября 2019

В настоящее время я построил двудольный график для данных о фондах и запасах. 

G = nx.from_pandas_edgelist(df, 'fund_code', 'stock_data', edge_attr='weight')
print(G.edges())

>>>
[('159957.SZ', '300070.SZ'), ('159957.SZ', '300027.SZ'), ('159957.SZ', '300003.SZ'), ('159958.SZ', '300017.SZ'), ('159958.SZ', '300136.SZ') ... ]

Затем я преобразовал график в смежную матрицу, чтобы выполнить некоторые вычисления. 

A = nx.adj_matrix(G)
print("getcol ", A.getrow(0))

>>>
getcol   
  (0, 1)    0.06296851574212893
  (0, 2)    0.06296851574212893
  (0, 3)    0.05247376311844078
  (0, 4)    0.20689655172413793
  ... ...

МойВопрос в том, как сопоставить узел 0, 1, 2 ... в matrix A с 159957.SZ', '300070.SZ' ... в graph G? Например, если у меня есть список [1, 3, 5, 2, 4], есть ли удобный способ преобразовать массив в [`159957.SZ`, `159959.SZ', `159995.SZ' ...]? Спасибо

...