如何用更少代码将Python字典元素按条件移至另一字典并清空原字典
如何用赋值表达式优化字典迁移的while循环代码
这是个代码优化练习,需求很明确:把一个Python字典中符合特定条件的元素迁移到另一个字典,无论有没有匹配元素,原字典必须被清空。
先看初始的示例数据和条件:
pie={"A1":{2000,2001,2002},"A2":{2003,2004,2005},"A3":{2000,2004,2007}} ; slices={} condition_check=2000
最初的while循环实现是这样的:
while bool(pie): temp = pie.popitem() if (condition_check in temp[1]): slices.update({temp[0]:temp[1]})
核心问题是:能不能去掉单独的temp = pie.popitem()这行,用赋值表达式把它整合到if条件里?
答案是可以的,利用Python 3.8及以上版本支持的赋值表达式(海象运算符:=),就能把赋值和条件判断合并到一行里,优化后的代码如下:
while pie: if condition_check in (item := pie.popitem())[1]: slices[item[0]] = item[1]
代码说明:
while pie和原代码的while bool(pie)效果完全一致,当pie为空时循环终止- 海象运算符
:=会先执行pie.popitem()弹出一个键值对元组,把它赋值给item变量,再用item[1]取到对应的值进行条件判断 - 如果条件满足,直接把
item里的键值对存入slices(用update({item[0]: item[1]})也可以,直接赋值更简洁) - 整个循环过程中,
pie.popitem()会逐个清空原字典,不需要额外调用clear()
你提到的推导式方案(比如字典推导后调用clear())确实也能实现需求,但如果执着于优化最初的while循环写法,上面这个用海象运算符的版本就是你要的答案。
内容的提问来源于stack exchange,提问作者danicotra
相关产品推荐
相关产品推荐

