Преобразовать словарь в массив Numpy - PullRequest
Новая
       7

Преобразовать словарь в массив Numpy

0 голосов
/ 03 января 2019

Я пытаюсь преобразовать словарь 

{0: {0: 173, 1: 342, 2: 666, 3: 506, 4: 94},
 1: {0: 13, 1: 2171, 2: 1915, 3: 3075, 4: 630},
 2: {0: 0, 1: 265, 2: 5036, 3: 508, 4: 11},
 3: {0: 0, 1: 3229, 2: 2388, 3: 3649, 4: 193},
 4: {0: 3, 1: 151, 2: 591, 3: 1629, 4: 410}}

в массив NumPy 

array([[ 173,  342,  666,  506,   94],
       [  13, 2171, 1915, 3075,  630],
       [   0,  265, 5036,  508,   11],
       [   0, 3229, 2388, 3649,  193],
       [   3,  151,  591, 1629,  410]])

Есть идеи, как это сделать эффективно?

1 Ответ

0 голосов
/ 03 января 2019

Цикл уровня Python здесь неизбежен, поэтому вы можете использовать понимание списка: 

res = np.array([list(item.values()) for item in d.values()])

# array([[ 173,  342,  666,  506,   94],
#        [  13, 2171, 1915, 3075,  630],
#        [   0,  265, 5036,  508,   11],
#        [   0, 3229, 2388, 3649,  193],
#        [   3,  151,  591, 1629,  410]])

Согласно комментарию @ FHTMitchell, предполагается, что элементы вашего словаря (внутренние и внешние) упорядочены надлежащим образом.Словари - это вставка, упорядоченная в 3.6 как деталь реализации CPython, и официально в 3.7 +.

Один из способов определить порядок для внутренних и внешних словарей - через operator.itemgetter: 

getter = itemgetter(*range(5))
res = np.array([getter(item) for item in getter(d)])

Такое решение не зависит от порядка вашего входного словаря.

...