将nd数组转换为键值字典

Question

在python中是否有一个函数将nd-array转换为字典,其中key是索引的元组,value是该索引处的矩阵值？

例如:

A = np.random.random([3,4,5])

结果:

{(i,j,k): A[i,j,k]}

Answer 1

当然:你可以np.ndenumerate用来完成这个:

{ index: v for index, v in np.ndenumerate(A) }

或简单地说,正如帝斯曼所指出的那样:

dict(np.ndenumerate(A))