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