在python上的一个函数中编写几个条件

ddrv8njm  于 2022-11-08  发布在  PyCharm
关注(0)|答案(1)|浏览(130)
def chocolate_maker(small, big, x):

     """
    :param small : small:is the num of the small chocolate bar chunks
    :param big : is the num of the big chocolate bar chunks
    :param x : total length of chocolate bar combining
    :return : returns true if the small+big == x or small*1== x or big*5 == x
    """
    if small * 1 + big * 5 == x or small * 1 == x or big * 5 == x:
        return True
    else:
        return False

print(chocolate_maker(3, 1, 3))

我需要对我的代码做一个简短的回顾,比如我是否可以让它更有效,比如可能拆分条件。

zbsbpyhn

zbsbpyhn1#

在不同的行上写条件可以增加可读性。但是这是一个简单的计算,你的版本足够高效。

def chocolate_maker(small: int, big: int, x: int) -> bool:

    return (
        small + big * 5 == x 
        or small == x 
        or big * 5 == x
    )

相关问题