在 Python 中獲取列表形狀

1. 用 Python 中的 len() 函式獲取一個列表的形狀
2. 用 Python 中的 numpy.shape() 方法獲取列表的形狀

在本教程中，我們將討論在 Python 中獲取列表形狀的方法。

用 Python 中的 len() 函式獲取一個列表的形狀

len() 函式為我們提供了一個物件中元素的數量。下面的程式碼示例向我們展示瞭如何在 Python 中使用 len(x) 方法來獲取一個列表的形狀。

lista = [[1, 2, 3], [4, 5, 6]]
arow = len(lista)
acol = len(lista[0])
print("Rows : " + str(arow))
print("Columns : " + str(acol))

輸出：

Rows : 2
Columns : 3

在上面的程式碼中，我們首先使用 len(lista) 得到 lista 中的行數，然後使用 len(lista[0]) 得到 lista 中的列數。

用 Python 中的 numpy.shape() 方法獲取列表的形狀

如果我們想讓我們的程式碼能夠處理任何多維列表，我們必須使用 numpy.shape() 方法。numpy.shape() 方法為我們提供了一個陣列中每個維度的元素數量。numpy.shape() 返回一個包含陣列每個維度元素數的元組。NumPy 包最初是設計用於陣列，但也可以用於列表。

NumPy 是一個外部包，並沒有預裝在 Python 中。我們需要在使用它之前安裝它。安裝 NumPy 包的命令如下。

pip install numpy

下面的程式碼示例顯示了我們如何使用 numpy.shape() 方法獲得一個列表的形狀。

import numpy as np

lista = [1, 2, 3, 4, 5]
listb = [[1, 2, 3], [4, 5, 6]]

print("Shape of lista is : " + str(np.shape(lista)))
print("Shape of listb is : " + str(np.shape(listb)))

輸出：

Shape of lista is : (5,)
Shape of listb is : (2, 3)

從上面的例子中可以看出，numpy.shape() 方法可以用於任何尺寸的列表。