在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，其键与原始字典result 相同，但值更少。

行：

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

报错

TypeError：列表索引必须是整数，而不是元组

我不明白，因为我可以创建 ex = result[result.keys()[0]][temp_v] 就好了。它只是不允许我使用 for 循环来执行此操作，以便我可以填充列表。

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

Answer 1

我不确定您要达到什么目的，但我可以解决您的问题：

np.where 正在返回一个元组，因此要访问您需要提供索引 temp_v[0]。此外，元组的值是一个数组，因此要遍历该值，您需要运行另一个循环 a for a in temp_v[0]，它可以帮助您访问该值。

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))