Cho hai ma trận nhập từ người dùng. Nhiệm vụ hiển thị phép cộng hai ma trận. Dùng nested list comprehension.
Thuật toán
Bước 1: Nhập hai ma trận. Bước 2: Nested for loops để lặp qua từng hàng và cột. Bước 3: Mỗi lần lặp cộng phần tử tương ứng từ hai ma trận và lưu kết quả.
Ví dụ mã
# Chương trình cộng hai ma trận dùng nested loop
A = []
n = int(input("Enter N for N x N matrix: "))
print("Enter the element ::>")
for i in range(n):
row = []
for j in range(n):
row.append(int(input()))
A.append(row)
print(A)
print("Display Array In Matrix Form")
for i in range(n):
for j in range(n):
print(A[i][j], end=" ")
print()
B = []
n = int(input("Enter N for N x N matrix: "))
print("Enter the element ::>")
for i in range(n):
row = []
for j in range(n):
row.append(int(input()))
B.append(row)
print(B)
print("Display Array In Matrix Form")
for i in range(n):
for j in range(n):
print(B[i][j], end=" ")
print()
result = [[0,0,0], [0,0,0], [0,0,0]]
for i in range(n):
for j in range(len(A[0])):
result[i][j] = A[i][j] + B[i][j]
print("Resultant Matrix is ::>")
for r in result:
print("Resultant Matrix is ::>", r)
Kết quả
Enter N for N x N matrix: 3 Enter the element ::> 10 10 10 20 20 20 30 30 30 [[10, 10, 10], [20, 20, 20], [30, 30, 30]] Display Array In Matrix Form 10 10 10 20 20 20 30 30 30 Enter N for N x N matrix: 3 Enter the element ::> 100 100 100 200 200 200 300 300 300 [[100, 100, 100], [200, 200, 200], [300, 300, 300]] Display Array In Matrix Form 100 100 100 200 200 200 300 300 300 Resultant Matrix is ::> [110, 110, 110] [220, 220, 220] [330, 330, 330]