获取 NumPy 中的行数

本教程将介绍如何获取 NumPy 数组的行数。

使用 array.shape 属性获取 NumPy 数组中的行数

NumPy 数组的 array.shape 属性 获取数组的形状。数组的形状表示数组每个维度中元素的数量。以下代码示例显示如何使用 array.shape 属性获取 2D NumPy 数组的行数和列数。

import numpy as np

array = np.array([[1,3,5],[2,4,6]])

rows, columns = array.shape

print("Rows = ",rows)
print("Columns = ", columns)

输出：

Rows =  2
Columns =  3

我们首先使用 np.array() 函数创建了一个 2D NumPy 数组。然后，我们使用 array.shape 属性确定数组的行数和列数，并将结果存储在 rows 和 columns 变量中。最后，我们将值打印在 rows 和 columns 中。