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

eye 函数是 NumPy 库中的一个函数，用于创建一个对角线上为1，其余元素为0的二维数组（也称为单位矩阵）

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