如何用数组拆分字典作为键的值

Question

我有一个包含 10 个值的字典，键为 1-10，每个键的值是维度 (126,1024) 的 ndarray。我想将此字典拆分为大小为 5 的块，因此结果将是两个大小为 5 的字典。我的代码如下：

pp=0
def chunks(data, SIZE=2):
    it = iter(data)
    for i in range(0, len(data), SIZE):
        yield {(i,k) for (i,k) in islice(it, SIZE)}

for item in chunks({(i,np.array(k)) for (i,k) in data_dic.items()}, 5):
    pp=pp+1
    dump_dic(item,pp)
    #print(item)

这段代码给了我一个 unhashable type: 'numpy.ndarray' 错误： for item in chunks({(i,np.array(k)) for (i,k) in data_dic.items()}, 5):

谁能建议我如何用 ndarray 数据拆分这本字典？

Answer 1

试试这个

dic = {1:np.zeros((20,20)), 2:np.zeros((20,20)), 3:np.zeros((20,20)), 4:np.zeros((20,20)), 5:np.zeros((20,20)), 6:np.zeros((20,20)), 7:np.zeros((20,20)), 8:np.zeros((20,20)), 9:np.zeros((20,20)), 10:np.zeros((20,20))}    
import math
num = 3 #number of dictionaries, in your question you said 2(5, 5),i added 3 dic (3, 3, 4) 
arr = []
for i in range(num):
    if i == num -1:
       arr.append(math.ceil(len(dic) / 3))
    else:        
       arr.append(math.floor(len(dic) / 3))


dic_list = []
m = 0
for i in arr:
   k = 1
   temp = {}
   for j in list(dic.values())[m:m+i]:
      temp[k] = j
      k += 1    
   dic_list.append(temp)    
   m = i  

# So you end up with array  of dictionaries. with 3, 3, 4 data

示例： 输入

{1:np.array(), 2:np.array(), 3:np.array(), 4:np.array(), 5:np.array(), 6:np.array(), 7:np.array(), 8:np.array(), 9:np.array(), 10:np.array()}

输出

  [{1:np.array(), 2:np.array(), 3:np.array()}, {4:np.array(), 5:np.array(), 6:np.array()}, {7:np.array(), 8:np.array(), 9:np.array(), 10:np.array()}]