Skip to content
Related Articles

Related Articles

Improve Article
Save Article
Like Article

Python – Rows with K string in Matrix

  • Last Updated : 11 Oct, 2020

Given a matrix, extract row numbers with particular string occurrence.

Input : test_list = [[“GFG”, “best”, “geeks”], [“geeks”, “rock”], [“GFG”, “for”, “CS”], [“Keep”, “learning”]], K = “GFG” 
Output : [0, 2] 
Explanation : 0th index and 2nd index have “GFG” in them as element.

Input : test_list = [[“GFG”, “best”, “geeks”], [“geeks”, “rock”, “GFG”], [“GFG”, “for”, “CS”], [“Keep”, “learning”]], K = “GFG” 
Output : [0, 1, 2] 
Explanation : 0th index, 1st index and 2nd index have “GFG” in them as element. 

Method #1 : Using loop

In this, we iterate for each element in Matrix and get the indices of all the rows which match the K string.

Python3




# Python3 code to demonstrate working of
# Rows with K string in Matrix
# Using loop
  
# initializing list
test_list = [["GFG", "best", "geeks"], ["geeks", "rock"],
             ["GFG", "for", "CS"], ["Keep", "learning"]]
  
# printing original list
print("The original list is : ", test_list)
  
# initializing K
K = "GFG"
  
res = []
  
# enumerate() used for getting both index and ele
for idx, ele in enumerate(test_list):
  
    # checking for K String
    if K in ele:
        res.append(idx)
  
# printing result
print("Rows with K : " + str(res))

Output:

The original list is :  [[‘GFG’, ‘best’, ‘geeks’], [‘geeks’, ‘rock’], [‘GFG’, ‘for’, ‘CS’], [‘Keep’, ‘learning’]]
Rows with K : [0, 2]

Method #2 : Using list comprehension

This is similar to above method, difference being its a shorthand to solve problem.

Python3




# Python3 code to demonstrate working of
# Rows with K string in Matrix
# Using list comprehension
  
# initializing list
test_list = [["GFG", "best", "geeks"], ["geeks", "rock"],
             ["GFG", "for", "CS"], ["Keep", "learning"]]
  
# printing original list
print("The original list is : ", test_list)
  
# initializing K
K = "GFG"
  
# shorthand to get result
res = [idx for idx, ele in enumerate(test_list) if K in ele]
  
# printing result
print("Rows with K : " + str(res))

Output:

The original list is :  [[‘GFG’, ‘best’, ‘geeks’], [‘geeks’, ‘rock’], [‘GFG’, ‘for’, ‘CS’], [‘Keep’, ‘learning’]]
Rows with K : [0, 2]


My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!