How to Print an Identity Matrix in Python

Submitted by razormist on Tuesday, June 24, 2025 - 22:30.

In this tutorial, we will learn how to program "How to Print an Identity Matrix in Python." The objective is to display an identity matrix based on the given number. This tutorial will guide you step by step through the process of generating an identity matrix. By the end of this tutorial, you will have a solid understanding of how to efficiently create identity matrices in Python, helping you strengthen your problem-solving and coding skills.

This topic is straightforward to understand. Just follow the instructions provided, and you will be able to complete it with ease. The program demonstrated will show you the correct and efficient way to display an identity matrix. So, let’s dive into the coding process!

Getting Started:

First you will have to download & install the Python IDLE's, here's the link for the Integrated Development And Learning Environment for Python https://www.python.org/downloads/.

Creating Main Function

This is the main function of the application. The following code will display a simple GUI in terminal console that will display program. To do this, simply copy and paste these blocks of code into the IDLE text editor.

while True:
    print("\n============== Print an Identity Matrix ==============\n")
 
    n=int(input("Enter a number: "))
    for i in range(0,n):
        for j in range(0,n):
            if(i==j):
                print("1",sep=" ",end=" ")
            else:
                print("0",sep=" ",end=" ")
        print()
 
    opt = input("\nDo you want to try again?(yes/no): ")
           
    if opt.lower() == 'yes':
        ret=False
    elif opt.lower() == 'no':
        ret=True
        print("Exiting program....")
    else:
        print("Please enter yes/no:")
        break
 
    if ret == False:
        continue
 
    

This Python program repeatedly prompts the user to enter a number n and prints an n x n identity matrix, where all diagonal elements are 1 and the rest are 0. The process continues in a loop until the user chooses to exit.

Output:

There you have it we successfully created How to Print an Identity Matrix in Python. I hope that this simple tutorial help you to what you are looking for. For more updates and tutorials just kindly visit this site. Enjoy Coding!

More Tutorials for Python Language

Python Tutorials