Python 代码重构:使用 any()和 all() 重构 for 循环

Drinkey 2020/11/3 5:03:48

问题描述使用 for 迭代一个可迭代对象,对每个对象进行条件判断,比如找到任何一个符合条件的元素,就返回 True,否则返回False。这种操作经常用于找符合条件的元素。代码片段 例1def is_user_exist(name: str) -> bool:for user in users:if user[user] == name:retu…

问题描述

使用 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]: True

In [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]: True

In [44]: is_user_exist_v2('johnny')
Out[44]: False

In [45]:

代码片段 例 2

from typing import List, Dict
def 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]: True

In [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]: True

In [62]: all_user_has_password_v2(users_group2)
Out[62]: False

总结

适当地使用 all()any() 可以有效减少代码的复杂度,能够很容易的对一个可迭代对象进行条件判断。

注,v2 的后缀是个非常不好的命名,在这里使用仅仅为了区分两个函数,不应该在代码中这样命名

参考

Built-in Functions — Python 3.7.7 documentation

随时随地学软件编程-关注百度小程序和微信小程序
关于找一找教程网

本站文章仅代表作者观点,不代表本站立场,所有文章非营利性免费分享。
本站提供了软件编程、网站开发技术、服务器运维、人工智能等等IT技术文章,希望广大程序员努力学习,让我们用科技改变世界。
[Python 代码重构:使用 any()和 all() 重构 for 循环]http://www.zyiz.net/tech/detail-148628.html

上一篇:安利一个Python大数据分析神器!

下一篇:阿里云对象存储OSS的python SDK示例

赞(0)

共有 条评论 网友评论

验证码: 看不清楚?
    关注微信小程序
    程序员编程王-随时随地学编程

    扫描二维码或查找【程序员编程王】

    可以随时随地学编程啦!

    技术文章导航 更多>
    扫一扫关注最新编程教程