使用 NumPy 查找给定矩阵的行数和列数

NumPy 是 Python 中用于科学计算的核心库之一，提供了丰富的数组操作和数学函数。在 NumPy 中，可以轻松地查找给定矩阵的行数和列数。

导入 NumPy 库

在开始之前，需要先导入 NumPy 库。可以使用以下命令来导入 NumPy：

import numpy as np

创建矩阵

在本例中，我们将使用 NumPy 来创建一个矩阵。可以使用以下命令来创建一个矩阵：

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

这将创建一个 3x3 的矩阵，如下所示：

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

查找行数和列数

可以使用 NumPy 中的 shape 属性来查找矩阵的行数和列数。shape 属性返回一个元组，其中第一个元素表示行数，第二个元素表示列数。

rows, cols = matrix.shape

上述代码将返回矩阵的行数和列数，并将它们分别赋值给变量 rows 和 cols。

完整代码

以下是一个完整的示例代码，演示如何使用 NumPy 查找给定矩阵的行数和列数。

import numpy as np

# 创建矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 查找行数和列数
rows, cols = matrix.shape

# 打印结果
print("矩阵的行数为:", rows)
print("矩阵的列数为:", cols)

以上代码的输出将是：

矩阵的行数为: 3
矩阵的列数为: 3

以上代码演示了如何使用 NumPy 查找给定矩阵的行数和列数。NumPy 不仅提供了丰富的数组操作和数学函数，还可以轻松地进行矩阵的操作和计算，是科学计算中不可或缺的工具之一。