python定義函數--參數檢查

參數檢查

調用函數時，如果參數個數不對，Python解釋器會自動檢查出來，並拋出TypeError：

>>> my_abs(1, 2)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: my_abs() takes 1 positional argument but 2 were given

但是如果參數類型不對，Python解釋器就無法幫我們檢查。試試my_abs和內置函數abs的差別：

>>> my_abs('A')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 2, in my_abs
TypeError: unorderable types: str() >= int()
>>> abs('A')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: bad operand type for abs(): 'str'

當傳入了不恰當的參數時，內置函數abs會檢查出參數錯誤，而我們定義的my_abs沒有參數檢查，會導致if語句出錯，出錯信息和abs不一樣。所以，這個函數定義不夠完善。

讓我們修改一下my_abs的定義，對參數類型做檢查，只允許整數和浮點數類型的參數。數據類型檢查可以用內置函數isinstance()實現：

def my_abs(x):
    if not isinstance(x, (int, float)):
        raise TypeError('bad operand type')
    if x >= 0:
        return x
    else:
        return -x

添加了參數檢查後，如果傳入錯誤的參數類型，函數就可以拋出一個錯誤：

>>> my_abs('A')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in my_abs
TypeError: bad operand type