溫馨提示×

python中shape函數(shù)的作用是什么

小億
236
2023-12-23 21:38:16
欄目: 編程語言

在python中,shape函數(shù)的作用是用來返回一個數(shù)組或矩陣的維度。對于一維數(shù)組,shape函數(shù)返回一個整數(shù),表示數(shù)組的長度。對于多維數(shù)組或矩陣,shape函數(shù)返回一個元組,表示各個維度的長度。

例如,對于一個一維數(shù)組arr,可以通過arr.shape獲取數(shù)組的長度。對于一個二維數(shù)組或矩陣arr,可以通過arr.shape獲取數(shù)組的行數(shù)和列數(shù)。

下面是一些示例:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
print(arr1.shape)  # 輸出 (5,)

arr2 = np.array([[1, 2, 3],
                 [4, 5, 6]])
print(arr2.shape)  # 輸出 (2, 3)

arr3 = np.array([[[1, 2, 3],
                  [4, 5, 6]],
                 [[7, 8, 9],
                  [10, 11, 12]]])
print(arr3.shape)  # 輸出 (2, 2, 3)

可以看到,shape函數(shù)返回的結(jié)果是一個元組,元組中的每個元素表示對應(yīng)維度的長度。對于一維數(shù)組,元組只有一個元素,表示數(shù)組的長度;對于多維數(shù)組或矩陣,元組的長度表示維度的個數(shù),每個元素表示對應(yīng)維度的長度。

0