eye函数在python中的错误处理方法

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 函数时进行适当的错误处理。