簡體   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