Python Dictionary value assignment with list comprehension

Question

I would like to take a python dictionary of lists, convert the lists to numpy arrays, and restore them in the dictionary using list comprehension.

For example, if I had a dictionary

myDict = {'A':[1,2,3,4], 'B':[5,6,7,8], 'C':'str', 'D':'str'}

I wish to convert the lists under keys A and B to numpy arrays, but leave the other parts of the dictionary untouched. Resulting in

myDict = {'A':array[1,2,3,4], 'B':array[5,6,7,8], 'C':'str', 'D':'str'}

I can do this with a for loop:

import numpy as np

for key in myDict:
    if key not in ('C', 'D'):
        myDict[key] = np.array(myDict[key])

But is it possible to do this with list comprehension? Something like

[myDict[key] = np.array(myDict[key]) for key in myDict if key not in ('C', 'D')]

Or indeed what is the fastest most efficient way to achieve this for a large dictionaries of long lists. Thanks, labjunky

The answers of Haidro and Ashwini Chaudhary, below, are very similar and appeared concurrently. — labjunky
– labjunky, Commented Aug 29, 2013 at 4:11

TerryA · Accepted Answer · 2013-08-28 06:58:58Z

3

With Python 2.7 and above, you can use a dictionary comprehension:

myDict = {'A':[1,2,3,4], 'B':[5,6,7,8], 'C':'str', 'D':'str'}
myDict = {key:np.array(val) if key not in {'C', 'D'} else val for key, val in myDict.iteritems()}

If you're below version 2.7 (and hence don't have dictionary comprehensions), you can do:

myDict = {'A':[1,2,3,4], 'B':[5,6,7,8], 'C':'str', 'D':'str'}
dict((key, np.array(val) if key not in {'C', 'D'} else val for key, val in myDict.iteritems())

edited Aug 28, 2013 at 6:58

answered Aug 28, 2013 at 6:42

TerryA

60.2k11 gold badges122 silver badges148 bronze badges

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

3 Comments

labjunky Over a year ago

Nice, but I get <ipython-input-135-01de37e639d5> in <dictcomp>((key, val)) ----> 1 myDict = {key:np.array(val) if val not in {'C', 'D'} else val for key, val in myDict.iteritems()} TypeError: unhashable type: 'list'

Andras Nemeth Over a year ago

Try ęlse key:val. It seems Python thinks in else val you want to use val as a key (that's why it says unhashable type.

TerryA Over a year ago

@Cucu No, it was a simple typo of me checking if the list was in the set, when I should have been checking if the value (which can't be unhashable) was in the set

Ashwini Chaudhary · Accepted Answer · 2013-08-28 06:55:44Z

1

To change all items except 'C' and 'D':

>>> myDict = {'A':[1,2,3,4], 'B':[5,6,7,8], 'C':'str', 'D':'str'}
>>> ignore = {'C', 'D'}
>>> new_dict = {k : v if k in ignore else np.array(v) for k,v in myDict.iteritems()}

the above dict-comprehension returns a new dictionary, to modify the original dict, try:

#myDict.viewkeys() - ignore --> set(['A', 'B'])
for key in myDict.viewkeys() - ignore:
    myDict[key] = np.array(myDict[key])

or if you only want to change 'A' and 'B':

for key in {'A', 'B'}:
    myDict[key] = np.array(myDict[key])

edited Aug 28, 2013 at 6:55

answered Aug 28, 2013 at 6:43

Ashwini Chaudhary

252k60 gold badges478 silver badges519 bronze badges

1 Comment

labjunky Over a year ago

spelling of 'ingore' second line

Greg · Accepted Answer · 2013-08-28 07:33:39Z

0

To

take a python dictionary of lists, convert the lists to numpy arrays, and restore them in the dictionary using list comprehension.

I would do:

myDict = {'A':[1,2,3,4], 'B':[5,6,7,8], 'C':'str', 'D':'str'}

def modifier(item):
    if type(item) == list:
        return np.array(item)
    else:
        return item

mod_myDict = {key: modifier(myDict[key]) for key in myDict.keys()}

The function then sets the restrictions you require in this case changing all lists into arrays. This returns:

{'A': array([1, 2, 3, 4]), 'B': array([5, 6, 7, 8]), 'C': 'str', 'D': 'str'}

Note: I believe this should be made shorter by using conditions in the if statement but I can't seem to figure it, something like

mod_myDict = {key: np.array(myDict[key]) if type(myDict[key])==list else key: myDict[key] for key in myDict.keys()}

That however, raises an error. Perhaps some more intelligent person than I knows why!

answered Aug 28, 2013 at 7:33

Greg

12.4k3 gold badges47 silver badges51 bronze badges

1 Comment

Brighid McDonnell Over a year ago

You may want something like the conditional comprehensions here: stackoverflow.com/questions/4260280/…

Collectives™ on Stack Overflow

Python Dictionary value assignment with list comprehension

3 Answers 3

3 Comments

1 Comment

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

3 Comments

1 Comment

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related