python - 如何从非零元素的索引字典创建稀疏 numpy 数组？

Question

我有一个与非零数字相对应的索引字典，并且想要编写一个函数来创建一个稀疏数组，该数组在给定索引处包含非零元素，中间有零。例如:

{(0,0):2, (1,1):3}

将输出到以下 numpy 数组

([[2,0], [0,3]])

将这个密集字典转换为稀疏数组的最简单方法是什么？

Answer 1

这应该可行，您唯一需要的就是，您应该知道输出的尺寸。

import numpy as np

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

S = 2
table = np.zeros((S,S))

for k,v in d.items():
    if d[k]:
        table[ k[0],k[1] ] = v

print(table)