Matrix creation of n*n in Python

When it is required to create a matrix of dimension n * n, a list comprehension is used.

Below is a demonstration of the same −

Example

 Live Demo

N = 4 print("The value of N is ") print(N) my_result = [list(range(1 + N * i, 1 + N * (i + 1)))    for i in range(N)] print("The matrix of dimension N * 0 is :") print(my_result)

Output

The value of N is 4 The matrix of dimension N * 0 is : [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]

Explanation

• The value of N is predefined.

• It is displayed on the console.

• It tells about the dimensions of the matrix.

• The number is iterated over, and is converted into a list.

• This is assigned to a variable.

• It is displayed on the console.