繁体   English   中英

用嵌套数组值转换字典

[英]transform dict with nested array values

我在下面有示例数据,这是默认词典。 我正在尝试将其转换为字典,如下面所需的输出,但是我在获取与原始字典中的键相匹配的键时遇到了麻烦。 我可以使用下面的代码获取值。 任何提示,不胜感激。

码:

res[0].item()

输出:

-1.613331913948059

示例数据:

res

defaultdict(<function __main__.<lambda>>,
            {0: array([-1.6133319], dtype=float32),
             1: array([-1.278326], dtype=float32),
             2: array([-0.68584293], dtype=float32),
             3: array([-1.2741858], dtype=float32),
             4: array([-0.81194735], dtype=float32)})

所需的输出:

{0: -1.6133319,
 1: -1.278326,
 2: -0.68584293,
 3: -1.2741858,
 4: -0.81194735}

IIUC,这个字典理解应该可以解决这个问题:

{k:v[0] for k,v in res.items()}

{0: -1.6133319, 1: -1.278326, 2: -0.68584293, 3: -1.2741858, 4: -0.81194735}

以下为我工作:

{k: res[k].item() for k in res}

这似乎已经成功了

test_res={}

for ids in list(res.keys()):
  test_res[ids]=res[ids].item()

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM