eye
函數是 NumPy 庫中的一個函數,用于創建一個對角線上為1,其余元素為0的二維數組(也稱為單位矩陣)
import numpy as np
def create_eye(n, m=None):
try:
if m is None:
return np.eye(n)
else:
return np.eye(n, m)
except TypeError as te:
print(f"TypeError: {te}")
return None
except ValueError as ve:
print(f"ValueError: {ve}")
return None
# 正確的使用方法
result = create_eye(3)
print(result)
result = create_eye(3, 4)
print(result)
# 錯誤的使用方法
result = create_eye("a")
print(result)
result = create_eye(3, "b")
print(result)
在這個示例中,我們定義了一個名為 create_eye
的函數,該函數接受兩個參數:n
和 m
。n
是必需的參數,表示矩陣的行數;m
是可選參數,表示矩陣的列數。如果未提供 m
,則默認值為 None
,此時將創建一個方陣。
在函數內部,我們使用 try-except
語句來捕獲可能發生的 TypeError
和 ValueError
異常。如果發生這些異常,我們將打印相應的錯誤信息并返回 None
。
通過這種方式,我們可以確保在調用 eye
函數時進行適當的錯誤處理。