Есть ли лучший, более короткий код для этого?

Question

Я пытаюсь изучить python, и один из сайтов, с которыми я работаю, имеет одну из следующих проблем: Учитывая два значения int, a и b, верните True, если один из них равен 6. Или если их сумма или разность равна 6.

и вот мой код он

def love6(a, b): 
    diff = a-b 
    sum = a+b 
    if a==6 or b==6: 
    return True 
    elif sum==6 or abs(diff)==6: 
    return True 
    else: 
    return False 

Это нормально, и работает отлично - просто интересно, если это может быть сжат дальше?

Answer 1

Поскольку две ветви вашего if сделать то же самое, вы можете объединить их в один, связывая их с другим or:

if a==6 or b==6 or sum==6 or abs(diff)==6: 
    return True 
else: 
    return False 

, но это просто:

return (a == 6 or b == 6 or sum==6 or abs(diff)==6) 

Но так как все из них испытывают одинаковое равенство, вы можете использовать оператор in следующим образом:

return 6 in (a,b,sum,abs(diff)) 

Стоит избегать имени переменной sum, так как есть встроенная функция с этим именем. Здесь, вы используете только один раз, и операция достаточно проста, так что вы можете оставить переменные полностью без каких-либо заметных потерь в ясности:

return 6 in (a,b,a+b,abs(a-b)) 

Answer 2

Зачем тратить время предварительного вычисления суммы & абсолютной разницы, когда они может не понадобиться?

def love6(a, b): 
    return a==6 or b==6 or a+b==6 or abs(a-b)==6 

питон or оператора короткого замыкания, поэтому вышеуказанные тесты оцениваются слева направо, а когда истинное значение найдено, не больше тестов выполняются. Например, если a или b равен 6, то сумма не рассчитывается, а абсолютная разница вычисляется только если ни a, ни b равно 6, а их сумма не 6.