Python 中矩陣是一種常見的數(shù)據(jù)類型,它由行和列組成。行表示矩陣中的橫向排列,而列則表示矩陣中的縱向排列。在 Python 中,我們可以使用 NumPy 庫來創(chuàng)建并處理矩陣。在這篇文章中,我們將介紹如何通過 Python 計算矩陣中行與列的和。
# 導(dǎo)入 NumPy 庫 import numpy as np # 創(chuàng)建一個 3 行 4 列的矩陣 matrix = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]) # 計算行的和 row_sum = matrix.sum(axis=1) # 計算列的和 column_sum = matrix.sum(axis=0) # 輸出結(jié)果 print("矩陣中每一行的和為:", row_sum) print("矩陣中每一列的和為:", column_sum)
在上面的代碼中,我們首先導(dǎo)入了 NumPy 庫。然后,我們使用 np.array 函數(shù)創(chuàng)建了一個 3 行 4 列的矩陣,其中每個元素的值從 1 到 12。接下來,我們分別使用 matrix.sum(axis=1) 和 matrix.sum(axis=0) 計算了矩陣中每一行和每一列的和,然后分別將它們存儲在 row_sum 和 column_sum 這兩個變量中。最后,我們通過 print 函數(shù)輸出了計算結(jié)果。
值得注意的是,在上面的代碼中,使用了 axis 參數(shù)來指定計算的方向。當我們將 axis 設(shè)置為 1 時,表示計算每一行的和;而將 axis 設(shè)置為 0 時,表示計算每一列的和。因此,在計算行和列的和時,我們需要根據(jù)具體需求指定正確的方向。