Python - Incremental K sized Row Matrix Initialization

Sometimes, while working with Python, we can have a problem in which we need to perform initialization of matrix with Incremental numbers. This kind of application can come in Data Science domain. Let’s discuss certain ways in which this task can be performed.

Method #1 : Using loop + list slicing
This task can be performed in brute way using loop. In this, we run a loop skipping by K (size of row required), to adjust numbers addition ahead.

# Python3 code to demonstrate  
# Incremental K sized Row Matrix Initialization 
# using loop + list slicing 
  
# Initialization of row size  
K = 3
  
# Incremental K sized Row Matrix Initialization 
# using loop + list slicing 
res = [] 
for idx in range(1, 10, K): 
    sub = [idx, idx + 1, idx + 2] 
    res.append(sub) 
  
# printing result  
print ("The Incremental Initialized Matrix is : " + str(res)) 

Output :

The Incremental Initialized Matrix is : [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Method #2 : Using list comprehension
This is just another way to perform this task. This performs in similar way as above, but in a shorter way.

# Python3 code to demonstrate  
# Incremental K sized Row Matrix Initialization 
# using list comprehension 
  
# Initialization of row size  
K = 3
  
# Incremental K sized Row Matrix Initialization 
# using list comprehension 
res = [[i, i + 1, i + 2] for i in range(1, 10, K)] 
  
# printing result  
print ("The Incremental Initialized Matrix is : " + str(res)) 

Output :

The Incremental Initialized Matrix is : [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

My Personal Notes

Recommended Posts:

manjeet_04

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.