Как преобразовать список смежных массивов в матрицу смежных массивов? - PullRequest
Новая
       57

Как преобразовать список смежных массивов в матрицу смежных массивов?

0 голосов
/ 24 октября 2018

У меня есть следующая таблица в массиве numpy.

enter image description here

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

1 Ответ

0 голосов
/ 24 октября 2018

Ваши данные более или менее в формате coo, поэтому используйте конструктор scipy.sparse.coo_matrix.Полученная разреженная матрица может быть преобразована во многие форматы.

Пример: 

>>> from pprint import pprint
>>> import numpy as np
>>> from scipy import sparse
>>> 
# create example
>>> a = np.random.randint(-10, 4, (10, 10)).clip(0, None) * 0.24
>>> a
array([[0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.24, 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.72, 0.24, 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.24, 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.24, 0.  , 0.72, 0.  , 0.48, 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.48, 0.24, 0.  ],
       [0.  , 0.  , 0.  , 0.48, 0.48, 0.  , 0.  , 0.24, 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.24, 0.48, 0.  , 0.  , 0.24],
       [0.  , 0.48, 0.  , 0.  , 0.  , 0.  , 0.72, 0.  , 0.  , 0.72],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ]])
>>>
# construct coo representation
>>> row, col = np.where(a)
>>> coo = np.rec.fromarrays([row, col, a[row, col]], names='row col value'.split())
>>> pprint(coo.tolist())
[(1, 2, 0.24),
 (2, 0, 0.72),
 (2, 1, 0.24),
 (3, 5, 0.24),
 (4, 2, 0.24),
 (4, 4, 0.72),
 (4, 6, 0.48),
 (5, 7, 0.48),
 (5, 8, 0.24),
 (6, 3, 0.48),
 (6, 4, 0.48),
 (6, 7, 0.24),
 (7, 5, 0.24),
 (7, 6, 0.48),
 (7, 9, 0.24),
 (8, 1, 0.48),
 (8, 6, 0.72),
 (8, 9, 0.72)]
>>>
# create sparse matrix
>>> out = sparse.coo_matrix((coo['value'], (coo['row'], coo['col'])), (10, 10))
>>> out
<10x10 sparse matrix of type '<class 'numpy.float64'>'
        with 18 stored elements in COOrdinate format>
>>> 
# convert back to dense format
>>> out.A
array([[0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.24, 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.72, 0.24, 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.24, 0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.24, 0.  , 0.72, 0.  , 0.48, 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.48, 0.24, 0.  ],
       [0.  , 0.  , 0.  , 0.48, 0.48, 0.  , 0.  , 0.24, 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.24, 0.48, 0.  , 0.  , 0.24],
       [0.  , 0.48, 0.  , 0.  , 0.  , 0.  , 0.72, 0.  , 0.  , 0.72],
       [0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  , 0.  ]])
...