使用Python的matrix()函数生成单位矩阵的示例

在Python中，可以使用NumPy库的matrix()函数来生成单位矩阵。单位矩阵是一个方阵，其中对角线上的元素都为1，其余元素都为0。单位矩阵的大小由用户指定。

下面是一个生成单位矩阵的示例代码：

import numpy as np

def generate_identity_matrix(size):
    # 生成一个给定大小的单位矩阵
    return np.matrix(np.eye(size))

# 示例使用
identity_matrix = generate_identity_matrix(3)
print(identity_matrix)

在上面的示例中，我们定义了一个名为generate_identity_matrix的函数，该函数接受一个参数size用于指定单位矩阵的大小。函数内部使用np.eye(size)生成对角线元素都为1的方阵，然后使用np.matrix()函数将其转换为NumPy矩阵。最后，函数返回生成的单位矩阵。

我们通过调用generate_identity_matrix(3)来生成一个3x3的单位矩阵，并将结果赋值给变量identity_matrix。然后，我们使用print()函数打印出该矩阵的值。

运行以上代码，将会输出以下结果：

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

正如预期的那样，输出结果是一个3x3的单位矩阵。对角线元素为1，其他元素都为0。

你可以根据需要调整参数size的值来生成不同大小的单位矩阵。例如，如果将参数size设置为4，将会生成一个4x4的单位矩阵。

identity_matrix = generate_identity_matrix(4)
print(identity_matrix)

输出结果为：

[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

可以看到，输出结果是一个4x4的单位矩阵。