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