獲取 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 中。