Python | n*n 的矩阵创建

在编程中，创建矩阵是十分常见的一项任务。本文将介绍如何在 Python 中创建 n*n 的矩阵。以下是几种方法：

1. 使用列表推导式

使用列表推导式可以非常方便地创建矩阵。只需要嵌套循环进行遍历，然后将每个元素加入到矩阵列表中即可。代码如下：

n = 3
matrix = [[0 for j in range(n)] for i in range(n)]
print(matrix)

输出：

[[0, 0, 0], [0, 0, 0], [0, 0, 0]]

如果需要将矩阵的元素初始化为其他值，只需要将 0 替换成相应的值即可。

2. 使用 NumPy 库

NumPy 是一个用于科学计算的 Python 库，它可以高效地处理各种数值运算。使用 NumPy 库创建矩阵非常简单，只需要使用 zeros 函数创建一个全零的矩阵，然后根据需要修改其中的元素即可。代码如下：

import numpy as np

n = 3
matrix = np.zeros((n, n))
print(matrix)

输出：

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

如果需要将矩阵的元素初始化为其他值，只需要将 zeros 函数中的参数替换为相应的值即可。

3. 使用矩阵库

如果希望能够更方便地进行矩阵相关的运算，可以使用一些专门的矩阵库，例如 NumPy、SciPy、SymPy 等。这些库中提供了丰富的矩阵操作接口，可以方便地进行各种操作。例如使用 NumPy 库创建一个 3*3 的矩阵：

import numpy as np

n = 3
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(matrix)

输出：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

总结

本文介绍了三种方法创建 n*n 的矩阵，分别是使用列表推导式、使用 NumPy 库和使用矩阵库。根据使用场景和需求选择不同的方法来创建矩阵，能够更加高效地完成编程任务。