Python Matrix Addition Program

Beginner
6 min read
Updated: Oct 2026
2 programs
Live preview

What Is Matrix Addition?

Matrix addition adds matching cells: C[i][j] = A[i][j] + B[i][j]. Both matrices must have the same number of rows and columns.

Remember
A + B is defined only when A and B have the same shape

  [ 1  2 ]   [ 5  6 ]   [ 6  8 ]
  [ 3  4 ] + [ 7  8 ] = [10 12 ]

Rule: (A + B)[i][j] = A[i][j] + B[i][j]
PositionABA + B
[0][0]156
[0][1]268
[1][0]3710
[1][1]4812

How to Add Matrices in Python

Four clear steps — nested loops do the work cell by cell.

  1. Confirm both matrices have the same rows and columns.
  2. Create a result matrix of that same size.
  3. Outer loop walks rows; inner loop walks columns.
  4. Set result[i][j] = A[i][j] + B[i][j] for every cell.
MethodIdeaTimeBest for
Nested loopsAdd matching cells into a new list of listsO(m × n)Interviews, teaching
Shape-safe addValidate len first, then addO(m × n)Safer production-style code

Pseudocode

Pseudocode
function AddMatrices(A, B):
    rows = rows of A
    cols = cols of A

    if rows != rows of B or cols != cols of B:
        error "shapes must match"

    C = new matrix[rows][cols]
    for i from 0 to rows - 1:
        for j from 0 to cols - 1:
            C[i][j] = A[i][j] + B[i][j]

    return C

Cheat Sheet

Keep these facts handy while writing the program.

TopicDetail
RuleC[i][j] = A[i][j] + B[i][j]
RequirementSame rows and same columns
Rows / colslen(A) / len(A[0])
LoopsOuter = rows, inner = columns
Not multiplyAddition is entrywise only
Classic check[[1,2],[3,4]] + [[5,6],[7,8]] = [[6,8],[10,12]]

Live Preview

Pick a sample pair to see matrices A, B, and A + B.

Tap a chip, then Add matrices (or tap a chip alone — it runs automatically).

Result
A =
1  2
3  4

B =
5  6
7  8

A + B =
6  8
10 12

Worked Walkthrough — 2×2

Visit every index once and add the matching cells.

Calculation
A = [1 2]    B = [5 6]
    [3 4]        [7 8]

i=0, j=0 → 1 + 5 = 6
i=0, j=1 → 2 + 6 = 8
i=1, j=0 → 3 + 7 = 10
i=1, j=1 → 4 + 8 = 12

Result = [6  8]
         [10 12]

Python Programs

Start with a clear nested-loop adder. Then add a shape check so mismatched sizes fail safely.

Example 1 — Nested Loops

Add two same-sized list-of-lists matrices cell by cell and print the result.

python
def add_matrices(a, b):
    rows = len(a)
    cols = len(a[0])
    result = [[0] * cols for _ in range(rows)]

    for i in range(rows):
        for j in range(cols):
            result[i][j] = a[i][j] + b[i][j]
    return result


def print_matrix(m):
    for row in m:
        print(" ".join(str(x) for x in row))


a = [[1, 2], [3, 4]]
b = [[5, 6], [7, 8]]
print_matrix(add_matrices(a, b))

How It Works

  1. Read rows with len(a) and columns with len(a[0]).
  2. Build a result matrix filled with zeros: [[0] * cols for _ in range(rows)].
  3. Outer loop chooses row i; inner loop chooses column j.
  4. Store a[i][j] + b[i][j] into result[i][j].
  5. Print walks each row and joins values with spaces.

Example 2 — Shape-Safe Add

Validate dimensions before adding so mismatched matrices fail with a clear error.

python
def add_matrices(a, b):
    rows = len(a)
    cols = len(a[0])

    if len(b) != rows or len(b[0]) != cols:
        raise ValueError("Matrices must have the same dimensions.")

    result = [[0] * cols for _ in range(rows)]
    for i in range(rows):
        for j in range(cols):
            result[i][j] = a[i][j] + b[i][j]
    return result


a = [[1, 2, 3], [4, 5, 6]]
b = [[7, 8, 9], [1, 1, 1]]
for row in add_matrices(a, b):
    print(" ".join(str(x) for x in row))

How It Works

  1. Compare both len(...) row counts and first-row column counts.
  2. Raise ValueError if shapes differ — addition would be undefined.
  3. Otherwise run the same nested-loop addition as Example 1.
  4. For the 2×3 sample: each cell sums independently.
  5. Mention this check in interviews even if the sample sizes already match.

Edge Cases & Pitfalls

Check these before calling the solution done.

Shape

Must match

Different sizes cannot be added — validate first.

Indexing

0-based

Loops run 0 .. rows-1 and 0 .. cols-1.

Rows

Keep rows even

Every row should have the same length — avoid ragged lists.

Multiply

Different problem

Do not confuse entrywise add with row-column multiply.

Time and Space Complexity

ProgramTimeExtra space
Nested loopsO(m × n)O(1) besides result
Shape-safe addO(m × n)O(1) besides result

Every cell is touched exactly once. The result matrix needs O(m × n) space to store the answer.

Key Takeaways

  • Rule: add matching cells; shapes must match.
  • Loops: outer rows, inner columns.
  • API: len(A) / len(A[0]).
  • Test: [[1,2],[3,4]] + [[5,6],[7,8]] → [[6,8],[10,12]].

One line: same shape required — then C[i][j] = A[i][j] + B[i][j].

Frequently Asked Questions

Only when they have the same dimensions — same row count and same column count. Then each result cell is the sum of matching cells.
Interview problems usually use a list of lists, such as [[1, 2], [3, 4]]. Each inner list is one row.
Yes. For equal-sized matrices, A + B equals B + A because each pair of matching entries is added.
No. Addition is entrywise. Multiplication mixes rows and columns and follows different dimension rules.
For an m × n matrix, each entry is visited once, so time is O(m × n). Extra space is O(1) besides the result matrix.

Next: Matrix Division

Divide matching cells of two matrices element-wise.

Matrix division tutorial →

About the author

Mari Selvan M P
Mari Selvan M P 🔗

Developer, cloud engineer, and technical writer

  • Experience 12 years building web and cloud systems
  • Focus Full Stack Development, AWS, and Developer Education

I write practical tutorials so students and working developers can learn by doing—from databases and APIs to deployment on AWS.

9 people found this page helpful