When it is required to find rows, which have ‘K’ string in matrix, the ‘enumerate’ attribute, a simple iteration and ‘append’ method is used.
Example
Below is a demonstration of the same −
my_list = [["Pyt", "fun", "python"], ["python", "rock"],["Pyt", "for", "CS"], ["Keep", "learning"]]
print("The list is :")
print(my_list)
K = "Pyt"
my_result = []
for idx, element in enumerate(my_list):
if K in element:
my_result.append(idx)
print("The result is :")
print(my_result)Output
The list is : [['Pyt', 'fun', 'python'], ['python', 'rock'], ['Pyt', 'for', 'CS'], ['Keep', 'learning']] The result is : [0, 2]
Explanation
A list is defined and displayed on the console.
The value for K is defined.
An empty list is created.
The list is iterated over using the ‘enumerate’ attribute.
A condition is placed, which checks if ‘K’ is present as one of the elements of the list.
If yes, its index is appended to the empty list.
This is the output that is displayed on the console.