提取for循环中的字典值以填充列表

Question

我有以下代码：

import numpy as np

result = {}
result['depth'] = [1,1,1,2,2,2]
result['generation'] = [1,1,1,2,2,2]
result['dimension'] = [1,2,3,1,2,3]
result['data'] = [np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0])]

for v in np.unique(result['depth']):
    temp_v = np.where(result['depth'] ==  v)
    values_v = [result[string][temp_v] for string in result.keys()]
    this_v = dict(zip(result.keys(), values_v))

在其中，我想创建一个名为“this_v”的新的dict”，其键与原始dict结果相同，但值较少。

该行：

values_v = [result[string][temp_v] for string in result.keys()]

给出一个错误

类型错误：列表索引必须是整数，而不是元组

我不明白，因为我可以创建ex=result[result.keys（）[0]][temp\u v]很好。它只是不允许我使用for循环来完成这项工作，这样我就可以填充列表。

知道为什么它不起作用吗？




Answer 1

匿名用户

我不确定您想要实现什么，但我可以解决您的问题：

np。其中返回一个元组，因此要访问，您需要给出索引temp_v[0]。此外，元组的值是一个数组，因此要循环该值，您需要在temp_v[0]中为a运行另一个循环a，这有助于您访问该值。

import numpy as np

result = {}
result['depth'] = [1,1,1,2,2,2]
result['generation'] = [1,1,1,2,2,2]
result['dimension'] = [1,2,3,1,2,3]
result['data'] = [np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0]), np.array([0,0,0])]

for v in np.unique(result['depth']):
    temp_v = np.where(result['depth'] ==  v)
    values_v = [result[string][a] for a in temp_v[0] for string in result.keys()]
    this_v = dict(zip(result.keys(), values_v))