Python如何按条件遍历字典键:指定具体键或传'all'遍历全部
解决方案
- 只需在循环内增加一个多条件判断,同时覆盖两种遍历需求即可:
- 传入
reg='all'时,所有字典键都满足判断条件 - 传入具体
reg值时,仅匹配到对应键时满足条件
- 传入
修改后的完整代码如下:
dic = {'a': 1, 'b': 2, 'c': 3} def iteration(reg=None): for k in dic.keys(): # 两个条件满足任意一个即可执行后续逻辑 if reg == 'all' or k == reg: # 这里替换为你实际要对匹配键执行的处理代码 print(k) # 测试调用 iteration(reg='a') # 仅处理键a iteration(reg='all') # 处理所有键
如果需要兼容reg默认是None的情况,避免未传参时执行无意义遍历,可以额外增加一层非空判断:
def iteration(reg=None): # 未传入有效参数直接返回 if reg is None: return for k in dic.keys(): if reg == 'all' or k == reg: print(k)
内容的提问来源于stack exchange,提问作者Janani Ethirraj
相关产品推荐
相关产品推荐

