在Python中,矩陣是一個二維數組。如果你編寫過許多代碼來操作矩陣,你很可能需要知道它們的維數是多少。Python中有一個內置的函數可以輕松地查看矩陣的維數。
>>>import numpy as np >>>a = np.array([[1, 2], [3, 4]]) >>>print(a.shape) (2, 2)
在這個例子中,我們創建了一個名為a的numpy數組,它由兩行和兩列組成。然后我們使用a.shape來查看它的維數,它返回一個元組(2, 2),這表明該矩陣有兩個行和兩個列。
如果你想獲取矩陣的行數或列數,你可以使用以下代碼:
>>>rows, cols = a.shape >>>print("Number of rows:", rows) Number of rows: 2 >>>print("Number of cols:", cols) Number of cols: 2
在這個例子中,我們使用a.shape來設置rows和cols,這是一個Python的 "解包 "操作,它將元組的第一個元素分配給rows變量,第二個元素分配給cols變量。然后我們使用print語句打印出行數和列數。
矩陣的維數對于在 Python 中進行矩陣運算和線性代數計算非常重要。因此,知道如何查看矩陣的維度是非常重要的。