矩陣是一個常見的線性代數工具,而在Python中使用矩陣也十分方便。本文將介紹如何在Python中獲取矩陣的一列。
# 導入NumPy庫
import numpy as np
# 創建一個3x3的矩陣
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 獲取矩陣的第一列
first_column = matrix[:, 0]
print(first_column) # 輸出[1 4 7]
在上述代碼中,我們使用了NumPy庫創建了一個3x3的矩陣。接著,我們可以使用Python的切片語法簡單地獲取到該矩陣的一列,例如:matrix[:, 0]
就代表矩陣的第一列,其中:
表示獲取所有行,0
則表示獲取第一列。
此時我們就成功地獲取了矩陣的一列。比如上述代碼中的first_column
就保存了矩陣的第一列,可以通過print
函數打印出來。