Python Program to Find Square of a Matrix

Examples:

Example1:

Input:

Given Matrix :
5  3  2
6  1  5
4  8  2
7  5  8

Output:

The Square of all elements of the given matrix is :
25 9 4 
36 1 25 
16 64 4 
49 25 64

Example2:

Input:

Given Matrix :
1  2  3
6  5  4
7  8  0

Output:

The Square of all elements of the given matrix is :
1 4 9 
36 25 16 
49 64 0

Program to Find Square of a Matrix in Python

Below are the ways to find the square of each element in the given matrix in Python:

• Using For Loop (Static Input)
• Using For loop (User Input)

Method #1: Using For Loop (Static Input)

Approach:

• Give the matrix as static input and store it in a variable.
• Calculate the number of rows of the given matrix by calculating the length of the nested list using the len() function and store it in a variable mtrxrows.
• Calculate the number of columns of the given matrix by calculating the length of the first list in the nested list using the len() function and store it in a variable mtrxcolums.
• Loop till the given number of rows using the For loop.
• Inside the For loop, Iterate till the given number of columns using another Nested For loop(Inner For loop).
• Print the square of the given element of the matrix by printing gvnmatrix[n][m]*gvnmatrix[n][m] value where n is the iterator value of the parent For loop and m is the iterator value of the inner For loop.
• The Exit of the Program.

Below is the implementation:

# Give the matrix as static input and store it in a variable.
mtrx = [[5, 3, 2], [6, 1, 5], [4, 8, 2], [7, 5, 8]]
# Calculate the number of rows of the given matrix by
# calculating the length of the nested list using the len() function
# and store it in a variable mtrxrows.
mtrxrows = len(mtrx)
# Calculate the number of columns of the given matrix by
# calculating the length of the first list in the nested list
# using the len() function and store it in a variable mtrxcols.
mtrxcols = len(mtrx[0])
print("The Square of all elements of the given matrix is :")
# To print all the elements of the given matrix.
# Loop till the given number of rows using the For loop.
for n in range(mtrxrows):
        # Inside the For loop, Iterate till the given number of columns using another
    # Nested For loop(Inner For loop).
    for m in range(mtrxcols):
      # Print the square of the given element of the matrix by printing gvnmatrix[n][m]*gvnmatrix[n][m]
      # value where n is the iterator value of the parent For loop and m is the iterator value
      # of the inner For loop.
        print(mtrx[n][m]*mtrx[n][m], end=' ')
    print()

Output:

The Square of all elements of the given matrix is :
25 9 4 
36 1 25 
16 64 4 
49 25 64

Method #2: Using For loop (User Input)

Approach:

• Give the number of rows of the matrix as user input using the int(input()) function and store it in a variable.
• Give the number of columns of the matrix as user input using the int(input()) function and store it in another variable.
• Take a list and initialize it with an empty value using [] or list() to say gvnmatrix.
• Loop till the given number of rows using the For loop
• Inside the For loop, Give all the row elements of the given Matrix as a list using the list(),map(),int(),split() functions and store it in a variable.
• Add the above row elements list to gvnmatrix using the append() function.
• To print all the elements of the given matrix.
• Loop till the given number of rows using the For loop.
• Inside the For loop, Iterate till the given number of columns using another Nested For loop(Inner For loop).
• Print the square of the given element of the matrix by printing gvnmatrix[n][m]*gvnmatrix[n][m] value where n is the iterator value of the parent For loop and m is the iterator value of the inner For loop.
• The Exit of the Program.

Below is the implementation:

# Give the number of rows of the matrix as user input using the int(input()) function
# and store it in a variable.
mtrxrows = int(input('Enter some random number of rows of the matrix = '))
# Give the number of columns of the matrix as user input using the int(input()) function
# and store it in another variable.
mtrxcols = int(input('Enter some random number of columns of the matrix = '))
# Take a list and initialize it with an empty value using [] or list() to say gvnmatrix.
mtrx = []
# Loop till the given number of rows using the For loop
for n in range(mtrxrows):
    # Inside the For loop, Give all the row elements of the given Matrix as a list using
    # the list(),map(),int(),split() functions and store it in a variable.
    l = list(map(int, input(
        'Enter {'+str(mtrxcols)+'} elements of row {'+str(n+1)+'} separated by spaces = ').split()))
    # Add the above row elements list to gvnmatrix using the append() function.

    mtrx.append(l)
    
print("The Square of all elements of the given matrix is :")
# To print all the elements of the given matrix.
# Loop till the given number of rows using the For loop.
for n in range(mtrxrows):
        # Inside the For loop, Iterate till the given number of columns using another
    # Nested For loop(Inner For loop).
    for m in range(mtrxcols):
      # Print the square of the given element of the matrix by printing gvnmatrix[n][m]*gvnmatrix[n][m]
      # value where n is the iterator value of the parent For loop and m is the iterator value
      # of the inner For loop.
        print(mtrx[n][m]*mtrx[n][m], end=' ')
    print()

Output:

Enter some random number of rows of the matrix = 3
Enter some random number of columns of the matrix = 3
Enter {3} elements of row {1} separated by spaces = 1 2 3
Enter {3} elements of row {2} separated by spaces = 6 5 4
Enter {3} elements of row {3} separated by spaces = 7 8 0
The Square of all elements of the given matrix is :
1 4 9 
36 25 16 
49 64 0

Practice Python Program Examples to master coding skills and learn the fundamental concepts in the dynamic programming language Python.

• C++ Program to Check if it is Sparse Matrix or Not
• Python Program to Find the Maximum Element in the Matrix
• Python Program to Find the Position of an Element in a Matrix
• Python Program to Read and Display a Matrix