จะค้นหาผลรวมของแถวและคอลัมน์ของเมทริกซ์ที่กำหนดโดยใช้ Numpy ได้อย่างไร

ในปัญหานี้ เราจะหาผลรวมของแถวทั้งหมดและคอลัมน์ทั้งหมดแยกกัน เราจะใช้ฟังก์ชัน sum() ในการหาผลรวม

อัลกอริทึม

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.

โค้ดตัวอย่าง

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)

ผลลัพธ์

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

คำอธิบาย

ฟังก์ชัน np.sum() ใช้เมทริกซ์เพิ่มเติมที่เรียกว่า 'แกน' Axis รับสองค่า 0 หรือ 1 ถ้า axis=0 จะบอกให้ฟังก์ชัน sum() พิจารณาเฉพาะแถวเท่านั้น ถ้า axis =1 จะเป็นการบอกฟังก์ชัน sum() ให้พิจารณาเฉพาะคอลัมน์เท่านั้น