功能的条件

时间:2013-10-10 03:41:43

标签: python

(Python 3.3.2)我必须写一个函数def numb(a,b,c,d,e),a,b,c,d,e的值只能是3,5或7 ,那么我怎么能写出这样一个条件,我试着将它们全部放在一行中将它们分开并且然后它不起作用...顺便说一下,用户将输入这些值,如果它们输入的数字不是3, 5,7然后我必须返回功能直到他们这样做(不要担心这部分,我知道该怎么做,我只是在编写条件时遇到了麻烦)

1 个答案:

答案 0 :(得分:5)

您可以将allgenerator expression

一起使用
def numb(a, b, c, d, e):
    assert all(x in {3, 5, 7} for x in (a, b, c, d, e))

另一种方法是使用sets

def numb(a, b, c, d, e):
    assert {a, b, c, d, e} <= {3, 5, 7}