Python – Filas con string K en Matrix

Dada una array, extraiga números de fila con ocurrencia de string particular.

Entrada : test_list = [[“GFG”, “best”, “geeks”], [“geeks”, “rock”], [“GFG”, “for”, “CS”], [“Keep”, “learning ”]], K = “GFG” 
Salida : [0, 2] 
Explicación : el índice 0 y el índice 2 tienen “GFG” como elemento.

Entrada : test_list = [[“GFG”, “best”, “geeks”], [“geeks”, “rock”, “GFG”], [“GFG”, “for”, “CS”], [“Keep ”, “aprendizaje”]], K = “GFG” 
Salida : [0, 1, 2] 
Explicación : el índice 0, el índice 1 y el índice 2 tienen “GFG” como elemento. 

Método #1: Usar bucle

En esto, iteramos para cada elemento en Matrix y obtenemos los índices de todas las filas que coinciden con la string K.

# 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))

Producción:

La lista original es: [[‘GFG’, ‘best’, ‘geeks’], [‘geeks’, ‘rock’], [‘GFG’, ‘for’, ‘CS’], [‘Keep’, ‘ aprendizaje’]]
Filas con K : [0, 2]

Método #2: Usar la comprensión de listas

Esto es similar al método anterior, la diferencia es que es una abreviatura para resolver el problema.

# 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))

Producción:

La lista original es: [[‘GFG’, ‘best’, ‘geeks’], [‘geeks’, ‘rock’], [‘GFG’, ‘for’, ‘CS’], [‘Keep’, ‘ aprendizaje’]]
Filas con K : [0, 2]