时间戳

首页 > 解决方案 > 重构多个相等检查的最pythonic方法

python - 重构多个相等检查的最pythonic方法

问题描述

给定一个 if 语句，如：

if response.status == SUCCESS or \
   response.status == FAILURE or \
   response.status == CLEAR or \
   response.status == READY:

像（1）这样重构是否更好：

if any(response.status == status for status in (SUCCESS, FAILURE, CLEAR, READY):

或（2）：

if response.status in {SUCCESS, FAILURE, CLEAR, READY}:

我的直觉是 1 更好，因为它更透明（如果不是更易读的话），但 2 更简洁，并且避免了遍历元组中的每个项目。

标签： pythonpython-3.xrefactoringzen-of-python

解决方案

您可以像这样使用元组或列表，无需迭代。(Python3.7)

>>>'SUCCESS' in ('SUCCESS', 'FAILURE', 'CLEAR', 'READY')
True

你的情况

if response.status in (SUCCESS, FAILURE, CLEAR, READY):

推荐阅读