Combine values from 2 dicts into a np.array python

Question

I have two dicts

a = {0:[1,2,3,4], 1:[5,6,7,8],...}
b = {0:[4,3,2,1], 1:[8,7,6,5],...}

I would like to create an np.array c for each key-value pair such as follows

c1 = array([[1,4],[2,3],[3,2],[4,1]])
c2 = array([[5,8],[6,7],[7,6],[8,5]])

How can I do this? Is it possible to store np.array in a python dict so that I can create a single dict c instead of multiple arrays

tobias_k · Accepted Answer · 2016-07-15 09:12:14Z

3

Yes, you can put np.array into a Python dictionary. Just use a dict comprehension and zip the lists from a and b together.

>>> a = {0:[1,2,3,4], 1:[5,6,7,8]}
>>> b = {0:[4,3,2,1], 1:[8,7,6,5]}
>>> c = {i: np.array(list(zip(a[i], b[i]))) for i in set(a) & set(b)}
>>> c
{0: array([[1, 4], [2, 3], [3, 2], [4, 1]]),
 1: array([[5, 8], [6, 7], [7, 6], [8, 5]])}

answered Jul 15, 2016 at 9:12

tobias_k

83.1k12 gold badges130 silver badges186 bronze badges

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

1 Comment

Raja Sattiraju Over a year ago

Wow. Thank you so much. Works elegantly

Colonel Beauvel · Accepted Answer · 2016-07-15 10:08:28Z

2

You can also use column_stack with a list comprehension:

import numpy as np

[np.column_stack((a[k], b[k])) for k in b.keys()]

Out[30]:
[array([[1, 4],
        [2, 3],
        [3, 2],
        [4, 1]]), array([[5, 8],
        [6, 7],
        [7, 6],
        [8, 5]])]

answered Jul 15, 2016 at 10:08

Colonel Beauvel

31.3k11 gold badges49 silver badges88 bronze badges

5 Comments

Raja Sattiraju Over a year ago

This method is almost twice as fast than using a dictionaray. Thank you so much

Colonel Beauvel Over a year ago

you are welcome! Certainly because I turned around zip (but you can use izip from itertools)

tobias_k Over a year ago

Does b.keys() guarantee that the keys come out in sorted (or any specific) order?

Colonel Beauvel Over a year ago

Answer is there: stackoverflow.com/questions/835092/…

tobias_k Over a year ago

@ColonelBeauvel That was not the question, but whether it's possible that the resulting array ends up as [c2, c1] instead of [c1, c2]. Using a dict with 100 keys being random numbers from 1 to 1000, keys() seems to return the keys in random order (Python 2 and 3).

Collectives™ on Stack Overflow

Combine values from 2 dicts into a np.array python

2 Answers 2

1 Comment

5 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

5 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related