[英]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.