Cách tìm tổng hàng và cột của ma trận cho trước bằng NumPy

Trong bài này, tìm tổng tất cả hàng và tất cả cột riêng biệt. Dùng hàm sum().

Thuật toán

Bước 1: Import numpy.
Bước 2: Tạo ma trận numpy mxn.
Bước 3: Lấy tổng tất cả hàng.
Bước 4: Lấy tổng tất cả cột.

Mã ví dụ

import numpy as np
a = np.matrix('10 20; 30 40')
print("Ma trận của ta: \n", a)
sum_of_rows = np.sum(a, axis=0)
print("\nTổng tất cả hàng: ", sum_of_rows)
sum_of_cols = np.sum(a, axis=1)
print("\nTổng tất cả cột: \n", sum_of_cols)

Kết quả

Ma trận của ta:  [[10 20]  [30 40]]
Tổng tất cả hàng:  [[40 60]]
Tổng tất cả cột:  [[30]  [70]]

Giải thích

np.sum() nhận tham số 'axis'. Axis=0: chỉ tính hàng. Axis=1: chỉ tính cột.