溫馨提示×

eye函數(shù)在python中的錯誤處理方法

小樊
84
2024-09-08 21:58:58
欄目: 編程語言

eye 函數(shù)是 NumPy 庫中的一個函數(shù),用于創(chuàng)建一個對角線上為1,其余元素為0的二維數(shù)組(也稱為單位矩陣)

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 的函數(shù),該函數(shù)接受兩個參數(shù):nmn 是必需的參數(shù),表示矩陣的行數(shù);m 是可選參數(shù),表示矩陣的列數(shù)。如果未提供 m,則默認值為 None,此時將創(chuàng)建一個方陣。

在函數(shù)內部,我們使用 try-except 語句來捕獲可能發(fā)生的 TypeErrorValueError 異常。如果發(fā)生這些異常,我們將打印相應的錯誤信息并返回 None。

通過這種方式,我們可以確保在調用 eye 函數(shù)時進行適當?shù)腻e誤處理。

0