Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để tìm tổng các hàng và cột của một ma trận nhất định bằng cách sử dụng Numpy?

Trong bài toán này, chúng ta sẽ tìm tổng của tất cả các hàng và tất cả các cột một cách riêng biệt. Chúng tôi sẽ sử dụng hàm sum () để lấy tổng.

Thuật toán

Step 1: Import numpy.
Step 2: Create a numpy matrix of mxn dimension.
Step 3: Obtain the sum of all the rows.
Step 4: Obtain the sum of all the columns.

Mã mẫu

import numpy as np

a = np.matrix('10 20; 30 40')
print("Our matrix: \n", a)

sum_of_rows = np.sum(a, axis = 0)
print("\nSum of all the rows: ", sum_of_rows)

sum_of_cols = np.sum(a, axis = 1)
print("\nSum of all the columns: \n", sum_of_cols)

Đầu ra

Our matrix:
 [[10 20]
 [30 40]]
Sum of all the rows:  [[40 60]]
Sum of all the columns:
 [[30]
 [70]]

Giải thích

Hàm np.sum () nhận thêm một ma trận được gọi là 'trục'. Axis nhận hai giá trị. 0 hoặc 1. Nếu axis =0, nó cho hàm sum () biết chỉ xem xét các hàng. Nếu axis =1, nó ra lệnh cho hàm sum () chỉ xem xét các cột.