具有列表推导的Python字典值赋值

Question

我想拿一个列表的python字典，将列表转换为numpy数组，并使用list comprehension在字典中恢复它们。

例如，如果我有一本字典

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

我希望将键A和B下的列表转换为numpy数组，但保持字典的其他部分不变。 导致

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

我可以使用for循环执行此操作：

import numpy as np

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

但是有可能用列表理解来做到这一点吗？ 就像是

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

或者实际上，对于长列表的大型词典来说，实现此目的的最快最有效的方法是什么。 谢谢，labjunky

Answer 1

使用Python 2.7及更高版本，您可以使用字典理解 ：

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

如果您的版本低于2.7（因此没有字典理解），您可以：

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

Answer 2

要更改除'C'和'D'以外'C'所有项目：

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

上面的dict-comprehension返回一个新的字典，修改原来的字典，试试：

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

或者如果你只想改变'A'和'B' ：

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

Answer 3

至

获取列表的python字典，将列表转换为numpy数组，并使用list comprehension在字典中恢复它们。

我会做：

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

然后，该函数设置在这种情况下所需的限制，将所有列表更改为数组。 返回：

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

注意：我相信这应该通过在if语句中使用条件来缩短，但我似乎无法想象它，类似于

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

然而，这引起了一个错误。 或许比我知道的更聪明的人为什么！