Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Chương trình Python cộng hai ma trận


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]