python多重字典转numpy_将带整数键的字典转换为numpy数组

让我们制作字典；这里我的obj是元组(为了方便)：In [563]: mydict={0:(0,),5:(1,),4:(3,),7:(4,)}

In [564]: mydict

Out[564]: {0: (0,), 4: (3,), 5: (1,), 7: (4,)}

初始化一个足够大的数组，dtype=object：

^{pr2}$

将值从mydict复制到A，使用键作为数组索引：In [567]: for k in mydict:

.....: A[k]=mydict[k]

.....:

In [568]: A

Out[568]: array([(0,), None, None, None, (3,), (1,), None, (4,)], dtype=object)

In [574]: A[[4,0]]

Out[574]: array([(3,), (0,)], dtype=object)

In [575]: A[[7,4]]

Out[575]: array([(4,), (3,)], dtype=object)

字典中定义的项现在出现在数组中相应的槽中。我不会说这个有用。在

我可以掩盖nones。在In [581]: Am=np.ma.masked_array(A)

In [582]: Am.mask=[False,True,True,True,False,False,True,False]

In [583]: Am

Out[583]:

masked_array(data = [(0,) (3,) (1,) (4,)],

mask = [False True True True False False True False],

fill_value = ?)

nones还在那里，只是“隐藏”了。我不知道蒙版对对象类型是否有用。在

dict子类

从注释来看，你最想要的是能够从字典中选择多个条目，类似于数组A[[0,3,5]]索引。在

子类dict可能比扩展或子类np.ndarray更容易。在

scipy.sparse有一个稀疏矩阵格式，它是dict的一个子类。它的__getitem__可能会给你如何扩展你自己的dict提供一些想法。我会设法想出一个更简单的版本。在

同时，获取一组密钥的一种方法是使用如下表达式：In [646]: {k:mydict[k] for k in mydict if k in {0,4}}

Out[646]: {0: (0,), 4: (3,)}

或者更简单In [647]: {k:mydict[k] for k in [0,4]}

Out[647]: {0: (0,), 4: (3,)}

但更强大：In [649]: {k:mydict.get(k,None) for k in [0,4,5,10]}

Out[649]: {0: (0,), 4: (3,), 5: (1,), 10: None}