如何以更Pythonic的方式拼接嵌套字典中的元素?
更Pythonic的嵌套字典元素拼接方案
假设我们有如下嵌套字典(由随机数生成):
import random nested_dict = { k: { 'val0': random.sample(range(100), 2), 'val1': random.sample(range(100), 2), } for k in 'abcd' }
要将所有内层字典中同名key对应的列表收集到一起,除了原有的双循环+defaultdict实现,这里提供几种更简洁优雅的Pythonic方案:
方案1:字典推导式 + 生成器表达式
如果确定所有内层字典的key完全一致,可以直接通过字典推导一次性构造目标字典:
# 先获取内层字典的key集合(取第一个内层字典即可) inner_keys = next(iter(nested_dict.values())).keys() dp_dict = {key: [inner[key] for inner in nested_dict.values()] for key in inner_keys} val0 = dp_dict['val0'] val1 = dp_dict['val1']
这种方式把循环逻辑压缩到推导式中,代码更紧凑,无需额外导入defaultdict。
方案2:利用zip和map批量处理
当内层字典结构完全统一时,可以用map提取所有内层字典,再通过zip配对对应key的取值:
# 提取所有内层字典 inner_dicts = nested_dict.values() # 批量获取每个key对应的列表集合 val0, val1 = zip(*(d.values() for d in inner_dicts)) # 如果需要列表类型而非元组,可转成list val0 = list(val0) val1 = list(val1)
这种方式更适合已知内层key数量和顺序的场景,代码极简。
关于Pandas方案的说明
正如你提到的,使用Pandas的实现虽然看似简洁,但对于这类小型字典操作来说,会引入不必要的库开销,效率远低于原生Python实现,因此不推荐。
内容的提问来源于stack exchange,提问作者itamar kanter
相关产品推荐
相关产品推荐

