এই প্রোবলেমে, আমরা সব সারি এবং সব কলামের যোগফল আলাদা আলাদা বের করব। এ জন্য sum() ফাংশন ব্যবহার করব।
এলগরিদম
ধাপ ১: numpy ইমপোর্ট করুন। ধাপ ২: mxn ডাইমেনশনের একটি numpy ম্যাট্রিক্স তৈরি করুন। ধাপ ৩: সব সারির যোগফল বের করুন। ধাপ ৪: সব কলামের যোগফল বের করুন।
উদাহরণ কোড
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 দুটি ভ্যালু নেয়: ০ বা ১। যদি axis=0 হয়, তবে এটি sum() ফাংশনকে শুধু সারিগুলো বিবেচনা করতে বলে। যদি axis=1 হয়, তবে এটি শুধু কলামগুলো বিবেচনা করতে বলে।