如何在Python中优雅遍历嵌套列表,基于阈值列表生成累计值并赋值
优化方案:简化嵌套列表遍历与阈值判断逻辑
核心思路
- 移除重复的逐个字典判断,改为遍历字典列表统一处理
- 跳过已完成赋值的字典,减少无效判断
- 保持逻辑清晰的同时,压缩冗余代码
原始代码示例(含重复逻辑)
list_of_lists = [[1, 2], [3, 4], [5, 6]] list_of_dicts = [{"vol": 5, "key": None}, {"vol": 20, "key": None}, {"vol": 40, "key": None}] total = 0 for sublist in list_of_lists: product = sublist[0] * sublist[1] total += product # 重复的判断逻辑,每个字典单独写if if total >= list_of_dicts[0]["vol"] and list_of_dicts[0]["key"] is None: list_of_dicts[0]["key"] = sublist[0] if total >= list_of_dicts[1]["vol"] and list_of_dicts[1]["key"] is None: list_of_dicts[1]["key"] = sublist[0] if total >= list_of_dicts[2]["vol"] and list_of_dicts[2]["key"] is None: list_of_dicts[2]["key"] = sublist[0]
优化后的基础版本
list_of_lists = [[1, 2], [3, 4], [5, 6]] list_of_dicts = [{"vol": 5, "key": None}, {"vol": 20, "key": None}, {"vol": 40, "key": None}] total = 0 for sublist in list_of_lists: total += sublist[0] * sublist[1] # 遍历字典列表,统一处理阈值判断 for d in list_of_dicts: if d["key"] is None and total >= d["vol"]: d["key"] = sublist[0] # 可选:所有字典完成赋值后提前终止循环 if all(d["key"] is not None for d in list_of_dicts): break
进一步优化(提升大数量场景效率)
如果字典数量较多,可维护一个待处理字典列表,只遍历未完成赋值的项:
list_of_lists = [[1, 2], [3, 4], [5, 6]] list_of_dicts = [{"vol": 5, "key": None}, {"vol": 20, "key": None}, {"vol": 40, "key": None}] total = 0 # 初始化待处理字典列表 pending_dicts = list_of_dicts.copy() for sublist in list_of_lists: total += sublist[0] * sublist[1] # 遍历切片副本,避免修改列表时打断遍历 for d in pending_dicts[:]: if total >= d["vol"]: d["key"] = sublist[0] pending_dicts.remove(d) # 无待处理项时提前退出 if not pending_dicts: break
优化点说明
- 消除重复判断:将每个字典的独立
if改为遍历循环,无论字典数量增减,代码结构无需调整 - 减少无效操作:通过
all()判断或维护待处理列表,提前终止循环,避免不必要的嵌套列表遍历 - 逻辑更易维护:核心逻辑集中在循环体内,后续修改阈值规则或字典结构时,只需调整一处判断条件
内容的提问来源于stack exchange,提问作者led_farmer
相关产品推荐
相关产品推荐

