配對的可選參數

Question

我有一個函數帶有兩個可選參數，我想要求兩者都可以，或者兩者都沒有指定。在Python中執行此操作的首選方式是什麼？我目前的做法似乎有點尷尬：

def myfunc(data, opt1=None, opt2=None): 
    if opt1 is None or opt2 is None: 
     if not (opt1 is None and opt2 is None): 
      raise ValueError("Must specify both opt1 and opt2 or neither.")  

這個問題：Python Optional Argument Pair是我的一樣，但在​​具體的上下文;我對通常在函數內執行此操作的默認方式感興趣。

Answer 1

你可以定義一個函數

def ensure_both_or_neither(opt1, opt2): 
    if opt1 is None or opt2 is None: 
     if not (opt1 is None and opt2 is None): 
      raise ValueError("Must specify both opt1 and opt2 or neither.") 

然後從你的函數調用它：

def myfunc(data, opt1=None, opt2=None): 
    ensure_both_or_neither(opt1, opt2) 

我不認爲這是任何比你一部開拓創新的方式更好，但它是稍微明確的/可讀？

Answer 2

def myfunc(data, *args): 
    if(len(args)==0 or len(args)==2): 
     #do something 

是我會怎麼做

這也使您可以爲您希望有儘可能多的參數，並且只有在使用沒有或2的代碼運行

Answer 3

簡化的條件你正在尋找的是：如果兩個參數不上的is None同一側擠在一起

if (opt1 is None) != (opt2 is None): 
    raise ... 

這種情況是真實的。

Answer 4

您可以使用這樣一個事實：set刪除重複：

def myfunc(data, opt1=None, opt2=None): 
    if len({opt1 is None, opt2 is None}) - 1: 
     raise ValueError("Must specify both opt1 and opt2 or neither.") 

你可以做任何數量的參數：

def myfunc(data, *args): 
    if len({arg is None for arg in args}) - 1: 
     raise ValueError("Must specify all parameters or none.") 

Answer 5

如果你既不需要或兩者兼而有之，不接受兩個獨立的論點;接受單個可選元組和文檔它。不要讓用戶有機會只提供一個。

def myfunc(data, opts=None): 
    """blah blah blah 

    opts, if given, should be a tuple with two values, neither None. 
    """ 
    opt1, opt2 = opts if opts is not None else (None, None) 

你可以把默認(None, None)，這將簡化你的函數（x1, x2 = opts）裏面的拆包，但塞滿您的函數的簽名爲代價。

（在現實中，opts可以是一個迭代有兩個元素，但如果用戶打電話像myfunc(data, "hi")，預計opt1是hi和opt2爲None，那是他的錯，不是你的。）