ในปัญหานี้ เราจะหาผลรวมของแถวทั้งหมดและคอลัมน์ทั้งหมดแยกกัน เราจะใช้ฟังก์ชัน sum() เพื่อหาผลรวม
อัลกอริทึม
ขั้นตอนที่ 1: นำเข้า numpy ขั้นตอนที่ 2: สร้าง Numpy Matrix ขนาด m x n ขั้นตอนที่ 3: หาผลรวมของแถวทั้งหมด ขั้นตอนที่ 4: หาผลรวมของคอลัมน์ทั้งหมด
ตัวอย่างโค้ด
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 Axis รับค่า 2 ค่า คือ 0 หรือ 1 หาก axis=0 จะบอกฟังก์ชัน sum() ให้พิจารณาเฉพาะแถว หาก axis=1 จะบอกให้พิจารณาเฉพาะคอลัมน์
