从由 i,j 标记的字典中的内容创建一个 numpy 矩阵

Question

假设我有一本字典：

d = {(0,0): 1, (0,1): 2, (1,0): 3, (1,1): 4}

我想把它放到数组中（即键对应于数组索引）

array([[1, 2],
       [3, 4]])

有没有比 for 循环更好的方法来做到这一点？ （我试过np.fromfunction但没有用。）

Answer 1

items给出了键和值的列表，您可以轻松地使用它来为预分配的数组分配值。

In [624]: d = {(0,0): 1, (0,1): 2, (1,0): 3, (1,1): 4}                                         
In [625]: list(d.items())                                                                      
Out[625]: [((0, 0), 1), ((0, 1), 2), ((1, 0), 3), ((1, 1), 4)]
In [626]: res = np.zeros((2,2),int)                                                            
In [627]: for id,val in d.items(): 
     ...:     res[id]=val 
     ...:                                                                                      
In [628]: res                                                                                  
Out[628]: 
array([[1, 2],
       [3, 4]])

或者使用keys和values列表分配元素：

In [640]: k,v = list(d.keys()), list(d.values())                                               
In [641]: res[tuple(np.transpose(k))]=v                                                        
In [642]: res                                                                                  
Out[642]: 
array([[1, 2],
       [3, 4]])

In [643]: np.transpose(k)                                                                      
Out[643]: 
array([[0, 0, 1, 1],
       [0, 1, 0, 1]])

Answer 2

似乎您有一个键情况字典，可以由scipy的dok_matrix处理：

from scipy.sparse import dok_matrix

d = {(0,0): 1, (0,1): 2, (1,0): 3, (1,1): 4}

mat = dok_matrix((2,2))
mat._update(d)
mat = mat.todense()

这给了你：

>>> mat
matrix([[1., 2.],
        [3., 4.]])

Answer 3

回来回答我自己的问题。 我遗漏的难题是 np.vectorize()。 看一下这个：

d = {(0,0): 1, (0,1): 2, (1,0): 3, (1,1): 4}
f = lambda i,j: d[(i,j)]
g = np.vectorize(f)
arr = np.fromfunction(f, (2,2))

这正是我想要的：

array([[1, 2],
       [3, 4]])

就这么简单。