首页 > 后端开发 > Python教程 > 如何在Python中根据自定义条件高效过滤字典?

如何在Python中根据自定义条件高效过滤字典?

DDD
发布: 2024-11-11 12:20:02
原创
780 人浏览过

How can I efficiently filter dictionaries based on custom conditions in Python?

使用自定义条件进行高级字典过滤

本书的示例说明了使用 items() 函数进行字典过滤,该过程可以简化为效率。

字典理解方法:

Python 为此目的提供了一个强大的工具:字典理解。有了它,您可以在应用自定义条件时根据现有值创建新字典。例如,要过滤点字典:

points = {'a': (3, 4), 'b': (1, 2), 'c': (5, 5), 'd': (3, 3)}

points_under_5 = {
    k: v
    for k, v in points.items()  # Iterate over (key, value) pairs
    if v[0] < 5 and v[1] < 5  # Filter based on condition
}
登录后复制

Python 2 兼容性:

在 Python 2(2.7 及以上版本)中,迭代字典项的语法略有不同:

points_under_5 = {
    k: v
    for k, v in points.iteritems()  # Use 'iteritems()' instead of 'items()'
    if v[0] < 5 and v[1] < 5
}
登录后复制

性能注意事项:

与手动迭代方法相比,字典理解方法提供了卓越的性能:

import timeit

# Manual iteration
manual_time = timeit.timeit(
    """
points={'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)}
points_small={}
for item in [i for i in points.items() if i[1][0]<5 and i[1][1]<5]:
    points_small[item[0]]=item[1]
""", number=1000000
)

# Dict comprehension
dict_time = timeit.timeit(
    """
points={'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)}
points_under_5 = {k:v for k, v in points.items() if v[0]<5 and v[1]<5}
""", number=1000000
)

print("Manual iteration time:", manual_time)
print("Dict comprehension time:", dict_time)
登录后复制

以上是如何在Python中根据自定义条件高效过滤字典?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板