获取 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 属性确定数组的行数和列数,并将结果存储在 rowscolumns 变量中。最后,我们将值打印在 rowscolumns 中。