我有一个大字典对象，它有几个键值对(大约16个)，但我只对其中3个感兴趣。将这样的字典子集化的最佳方法(最短/有效/最优雅)是什么?
我知道的最好的是:
bigdict = {'a':1,'b':2,....,'z':26}
subdict = {'l':bigdict['l'], 'm':bigdict['m'], 'n':bigdict['n']}
我相信有比这更优雅的方式。
当前回答
此答案使用与所选答案类似的字典推导，但不会对缺失项进行省略。
Python 2版本:
{k:v for k, v in bigDict.iteritems() if k in ('l', 'm', 'n')}
Python 3版本:
{k:v for k, v in bigDict.items() if k in ('l', 'm', 'n')}
其他回答
此答案使用与所选答案类似的字典推导，但不会对缺失项进行省略。
Python 2版本:
{k:v for k, v in bigDict.iteritems() if k in ('l', 'm', 'n')}
Python 3版本:
{k:v for k, v in bigDict.items() if k in ('l', 'm', 'n')}
解决方案
from operator import itemgetter
from typing import List, Dict, Union
def subdict(d: Union[Dict, List], columns: List[str]) -> Union[Dict, List[Dict]]:
"""Return a dict or list of dicts with subset of
columns from the d argument.
"""
getter = itemgetter(*columns)
if isinstance(d, list):
result = []
for subset in map(getter, d):
record = dict(zip(columns, subset))
result.append(record)
return result
elif isinstance(d, dict):
return dict(zip(columns, getter(d)))
raise ValueError('Unsupported type for `d`')
使用实例
# pure dict
d = dict(a=1, b=2, c=3)
print(subdict(d, ['a', 'c']))
>>> In [5]: {'a': 1, 'c': 3}
# list of dicts
d = [
dict(a=1, b=2, c=3),
dict(a=2, b=4, c=6),
dict(a=4, b=8, c=12),
]
print(subdict(d, ['a', 'c']))
>>> In [5]: [{'a': 1, 'c': 3}, {'a': 2, 'c': 6}, {'a': 4, 'c': 12}]
如果有人想要字典的前几项n而不知道键:
n = 5 # First Five Items
ks = [*dikt.keys()][:n]
less_dikt = {i: dikt[i] for i in ks}
py3.8+中另一种避免big_dict中缺少键的None值的方法使用walrus:
small_dict = {key: val for key in ('l', 'm', 'n') if (val := big_dict.get(key))}
如果你想保留大部分键，同时删除一些键，另一种方法是:
{k: bigdict[k] for k in bigdict.keys() if k not in ['l', 'm', 'n']}
