问题形容
应用 for 迭代一个可迭代对象,对每个对象进行条件判断,比方找到任何一个符合条件的元素,就返回 True
,否则返回False
。这种操作常常用于找符合条件的元素。
代码片段 例1
def is_user_exist(name: str) -> bool: for user in users: if user['user'] == name: return True return False
运行:判断用户是否存在
In [33]: users = [{'user':'john','password':'mypassword'}, {'user':'william', 'password':'yourpass'}]In [38]: is_user_exist('john')Out[38]: TrueIn [39]: is_user_exist('johny')Out[39]: False
重构
应用 any(iterable)
重构
def is_user_exist_v2(name: str) -> bool: return any(user['user'] == name for user in users)
运行
In [43]: is_user_exist_v2('john')Out[43]: TrueIn [44]: is_user_exist_v2('johnny')Out[44]: FalseIn [45]:
代码片段 例 2
from typing import List, Dictdef all_user_has_password(usergroup: List[Dict]) -> bool: for user in usergroup: if user['password'] == '': return False return True
运行:判断用户组是否都设置了明码
In [49]: users_group1 = [{'user':'john','password':'mypassword'}, {'user':'william', 'password':'yourpass'}]In [50]: users_group2 = [{'user':'john','password':'mypassword'}, {'user':'william', 'password':'yourpass'}, {'user': 'user1', 'password':''}]In [58]: all_user_has_password(users_group1)Out[58]: TrueIn [59]: all_user_has_password(users_group2)Out[59]: False
重构
应用 all(iterable)
进行重构
def all_user_has_password_v2(usergroup: List[Dict]) -> bool: return all(user['password'] != '' for user in usergroup)
运行
In [61]: all_user_has_password_v2(users_group1)Out[61]: TrueIn [62]: all_user_has_password_v2(users_group2)Out[62]: False
总结
适当地应用 all()
和 any()
能够无效缩小代码的复杂度,可能很容易的对一个可迭代对象进行条件判断。
注,v2 的后缀是个十分不好的命名,在这里应用仅仅为了辨别两个函数,不应该在代码中这样命名
参考
Built-in Functions — Python 3.7.7 documentation