Python: use dict and list of tuples to create a numpy array

Question

I have five key/value pairs in a python dict object:

dict1 = {0:3, 1:2, 2:2, 3:2, 4:3}

And I have a list of tuples:

list_of_tuples = [(0, 1), (0, 4), (2, 3)]

I want a numpy array of shape (5, 5, 2) that contains the value from the dict object for each tuple value. I get 5 from the number of keys in the dict object, and 2 from the number of values in each tuple.

That is:

at position (0, 1) of the numpy array, I want a numpy array with values (3, 2);
at position (0, 4) of the numpy array, I want a numpy array with values (3, 3);
at position (2, 3) of the numpy array, I want a numpy array with values (2, 2);
at all other positions of the numpy array, I want a numpy array with values (4, 4),

where I use (4, 4) as an indicator of an empty spot.

Is there a nice, Pythonic way of doing this?

Can you do this in some non-nice Pythonic way? You know, with ordinary dictionary access and indexing? — hpaulj
– hpaulj, Commented Feb 3, 2018 at 17:59

Alex · Accepted Answer · 2018-02-03 17:46:09Z

2

You can create an array of 4s using np.full and then populate it:

a = np.full((5, 5, 2), 4)
for x, y in list_of_tuples:
     a[x, y] = dict1[x], dict1[y]

answered Feb 3, 2018 at 17:46

Alex

19.2k9 gold badges65 silver badges82 bronze badges

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

Collectives™ on Stack Overflow

Python: use dict and list of tuples to create a numpy array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related