在 Python 中獲取列表形狀
Muhammad Maisam Abbas
2023年1月30日
2021年2月28日
在本教程中,我們將討論在 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()
方法可以用於任何尺寸的列表。
Author: Muhammad Maisam Abbas
Maisam is a highly skilled and motivated Data Scientist. He has over 4 years of experience with Python programming language. He loves solving complex problems and sharing his results on the internet.
LinkedIn