transform dict with nested array values

Question

I have the example data below, which is a default dictionary. I’m trying to transform it in to a dictionary like the desired output below, but I’m having trouble getting the keys to match the keys from the original dictionary. I can get the values using the code below. Any tips are greatly appreciated.

Code:

res[0].item()

Output:

-1.613331913948059

Example data:

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

Desired output:

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

sacuL · Accepted Answer · 2018-08-08 22:37:42Z

1

IIUC, This dictionary comprehension should do the trick :

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

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

answered Aug 8, 2018 at 22:37

sacuL

51.6k9 gold badges88 silver badges115 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user3476463 Over a year ago

thanks for getting back to me so quickly, that worked.

rafaelc · Accepted Answer · 2018-08-08 23:53:34Z

1

The following worked for me:

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

edited Aug 8, 2018 at 23:53

rafaelc

59.4k15 gold badges64 silver badges87 bronze badges

answered Aug 8, 2018 at 22:42

Mashume

338 bronze badges

Comments

user3476463 · Accepted Answer · 2018-08-08 22:38:10Z

0

This seems to have done the trick

test_res={}

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

answered Aug 8, 2018 at 22:38

user3476463

4,61524 gold badges77 silver badges146 bronze badges

1 Comment

rafaelc Over a year ago

Even though this works, it will be slower than a dict comprehension

Collectives™ on Stack Overflow

transform dict with nested array values

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related